Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix ruff error

Browse Source
This commit is contained in:
Lincoln Stein 2024-05-17 22:53:45 -04:00
parent 8aebc29b91
commit e77c7e40b7
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/app/services/model_load/test_load_api.py
View File

@ -79,4 +79,6 @@ def test_download_diffusers(mock_context: InvocationContext) -> None:
def test_download_diffusers_subfolder(mock_context: InvocationContext) -> None: def test_download_diffusers_subfolder(mock_context: InvocationContext) -> None:
model_path = mock_context.models.download_and_cache_model("stabilityai/sdxl-turbo::vae") model_path = mock_context.models.download_and_cache_model("stabilityai/sdxl-turbo::vae")
assert model_path.is_dir() assert model_path.is_dir()
assert (model_path / "diffusion_pytorch_model.fp16.safetensors").exists() or (model_path / "diffusion_pytorch_model.safetensors").exists() assert (model_path / "diffusion_pytorch_model.fp16.safetensors").exists() or (
model_path / "diffusion_pytorch_model.safetensors"
).exists()