aljaz/generative-models
1
0
Fork 0
mirror of https://github.com/Stability-AI/generative-models.git synced 2026-02-02 20:34:27 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix checkpoint loading too

Browse Source
This commit is contained in:
Stephan Auerhahn
2023-08-03 17:56:24 -07:00
parent 84d3a7f6f5
commit 4aea6fa2a4
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
sgm/inference/api.py
View File

@@ -178,6 +178,9 @@ class SamplingPipeline:
if model_path is None:
model_path = pathlib.Path(__file__).parent.parent.resolve() / "checkpoints"
if not os.path.exists(model_path):
# This supports development installs where checkpoints is root level of the repo
model_path = pathlib.Path(__file__).parent.parent.parent.resolve() / "checkpoints"
if config_path is None:
config_path = (
pathlib.Path(__file__).parent.parent.resolve() / "configs/inference"