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

fix fallback logic for config path

Browse Source
This commit is contained in:
Stephan Auerhahn
2023-08-03 17:50:10 -07:00
parent 19fa4da3de
commit 84d3a7f6f5
1 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
sgm/inference/api.py
View File

@@ -182,15 +182,12 @@ class SamplingPipeline:
config_path = (
pathlib.Path(__file__).parent.parent.resolve() / "configs/inference"
)
if not os.path.exists(config_path):
# This supports development installs where configs is root level of the repo
config_path = pathlib.Path(__file__).parent.parent.parent.resolve() / "configs/inference"
self.config = str(config_path / self.specs.config)
self.ckpt = str(model_path / self.specs.ckpt)
if not os.path.exists(self.config):
# This supports development installs where configs is root level of the repo
if config_path is None:
config_path = (
pathlib.Path(__file__).parent.parent.parent.resolve()
/ "configs/inference"
)
raise ValueError(
f"Config {self.config} not found, check model spec or config_path"
)