Skip to content

Commit 5767112

Browse files
committed
+ tests
1 parent f9b317c commit 5767112

File tree

7 files changed

+113
-86
lines changed

7 files changed

+113
-86
lines changed

tests/python_tests/requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ pytest==9.0.0
77
transformers==4.55.4
88
hf_transfer==0.1.9
99
gguf==0.17.1
10+
opencv-python==4.12.0.88
1011

1112
# rag requirements
1213
langchain_community==0.4.1

tests/python_tests/samples/test_tools_llm_benchmark.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,12 @@
2727
{"steps": 30, "width": 64, "height": 128, "guidance_scale": 1.0, "strength": "0.8", "media": "cat.png", "prompt": image_generation_i2i_prompt},
2828
]
2929

30+
# @pytest.fixture(scope="module")
31+
# def real_video(pytestconfig):
32+
# spinning_earth_url = TEST_VIDEO_URLS["spinning-earth-480"]
33+
# video = from_cache_or_download(pytestconfig, spinning_earth_url, "spinning-earth-480.mp4")
34+
# return video
35+
3036
class TestBenchmarkLLM:
3137
@pytest.mark.samples
3238
@pytest.mark.parametrize(

0 commit comments

Comments
 (0)