Skip to content

Commit

Permalink
revert test
Browse files Browse the repository at this point in the history
  • Loading branch information
Wovchena committed May 28, 2024
1 parent 6856b5e commit 5b5fd01
Showing 1 changed file with 0 additions and 8 deletions.
8 changes: 0 additions & 8 deletions tests/python_tests/test_generate_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,14 +54,6 @@ def stop_criteria_map():
# (dict(num_beam_groups=2, num_beams=8, num_return_sequences=8, max_new_tokens=20, diversity_penalty=1.0), 'The Sun is yellow because'),
# (dict(num_beam_groups=2, num_beams=8, num_return_sequences=8, max_new_tokens=20, diversity_penalty=1.5), 'The Sun is yellow because'),
]


def test_short():
import openvino_genai
openvino_genai.LLMPipeline(r'C:\Users\vzlobin\r\g\TinyLlama-1.1B-Chat-v1.0', 'CPU', {})
openvino_genai.LLMPipeline(r'C:\Users\vzlobin\r\g\TinyLlama-1.1B-Chat-v1.0', 'CPU', {}, '')


@pytest.mark.parametrize("generation_config,prompt", test_cases)
def test_greedy_decoding(model_fixture, generation_config, prompt):
run_hf_ov_genai_comparison(model_fixture, generation_config, prompt)
Expand Down

0 comments on commit 5b5fd01

Please sign in to comment.