Fix path logic for development installs

Stability-AI · Aug 4, 2023 · 4e2236f · 4e2236f
1 parent baf79d2
commit 4e2236f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/sgm/inference/api.py b/sgm/inference/api.py
@@ -185,6 +185,11 @@ def __init__(
         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"
             )