wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-09 21:12:36 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

Tweak test

Browse Source
This commit is contained in:
doctorpangloss 2024-09-11 14:47:23 -07:00
parent 25e636fb65
commit 3548fcd3c2
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/unit/test_language_nodes.py
View File

@ -75,7 +75,9 @@ def mock_openai_client():
yield instance yield instance
def test_transformers_loader(): def test_transformers_loader(has_gpu):
if not has_gpu:
pytest.skip("requires GPU")
loader = TransformersLoader() loader = TransformersLoader()
model, = loader.execute("microsoft/Phi-3-mini-4k-instruct", "") model, = loader.execute("microsoft/Phi-3-mini-4k-instruct", "")
assert isinstance(model, LanguageModel) assert isinstance(model, LanguageModel)