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

add ram cache module and support files

Browse Source
This commit is contained in:
Lincoln Stein
2024-01-31 23:37:59 -05:00
committed by psychedelicious
parent 010c4eae65
commit b8e875bb73
14 changed files with 1334 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test_model_probe.py
View File

@ -21,9 +21,10 @@ def test_get_base_type(vae_path: str, expected_type: BaseModelType, datadir: Pat
base_type = probe.get_base_type()
assert base_type == expected_type
repo_variant = probe.get_repo_variant()
assert repo_variant == 'default'
assert repo_variant == "default"
def test_repo_variant(datadir: Path):
probe = VaeFolderProbe(datadir / "vae" / "taesdxl-fp16")
repo_variant = probe.get_repo_variant()
assert repo_variant == 'fp16'
assert repo_variant == "fp16"