Fix path logic for development installs

This commit is contained in:
Stephan Auerhahn
2023-08-04 00:15:22 +00:00
parent baf79d2d79
commit 4e2236f67d

View File

@@ -185,6 +185,11 @@ class SamplingPipeline:
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"
)