add support for generic loading of diffusers directories

This commit is contained in:
Lincoln Stein
2024-06-03 20:31:05 -04:00
committed by psychedelicious
parent a9962fd104
commit f81b8bc9f6
8 changed files with 44 additions and 27 deletions

View File

@ -60,6 +60,10 @@ def mm2_model_files(tmp_path_factory) -> Path:
def embedding_file(mm2_model_files: Path) -> Path:
return mm2_model_files / "test_embedding.safetensors"
@pytest.fixture
def vae_directory(mm2_model_files: Path) -> Path:
return mm2_model_files / "taesdxl"
@pytest.fixture
def diffusers_dir(mm2_model_files: Path) -> Path: