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

fixup cachedir call

Browse Source
This commit is contained in:
Lincoln Stein 2023-05-18 14:52:16 -04:00
parent a77c4c87b2
commit 259d6ec90d
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
invokeai/backend/model_management/model_cache.py
View File

@ -232,7 +232,7 @@ class DiffusersModelInfo(ModelInfoBase):
model = self.child_types[child_type].from_pretrained(
self.repo_id_or_path,
subfolder=child_type.value,
cache_dir=get_invokeai_config.cache_dir('hub'),
cache_dir=get_invokeai_config().cache_dir,
torch_dtype=torch_dtype,
variant=variant,
)
@ -347,7 +347,7 @@ class ClassifierModelInfo(ModelInfoBase):
model = self.child_types[child_type].from_pretrained(
self.repo_id_or_path,
subfolder=child_type.value,
cache_dir=get_invokeai_config().cache_dir('hub'),
cache_dir=get_invokeai_config().cache_dir,
torch_dtype=torch_dtype,
)
# calc more accurate size
@ -392,7 +392,7 @@ class VaeModelInfo(ModelInfoBase):
model = self.vae_type.from_pretrained(
self.repo_id_or_path,
cache_dir=get_invokeai_config().cache_dir('hub'),
cache_dir=get_invokeai_config().cache_dir,
torch_dtype=torch_dtype,
)
# calc more accurate size