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

black

Browse Source
This commit is contained in:
Stephan Auerhahn
2023-08-04 00:00:51 +00:00
parent 44943df4f2
commit baf79d2d79
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
sgm/inference/api.py
View File

@@ -165,7 +165,7 @@ class SamplingPipeline:
config_path=None,
device="cuda",
use_fp16=True,
) -> None:
) -> None:
self.model_id = model_id
if model_spec is not None:
self.specs = model_spec
@@ -179,13 +179,19 @@ class SamplingPipeline:
if model_path is None:
model_path = pathlib.Path(__file__).parent.parent.resolve() / "checkpoints"
if config_path is None:
config_path = pathlib.Path(__file__).parent.parent.resolve() / "configs/inference"
config_path = (
pathlib.Path(__file__).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):
raise ValueError(f"Config {self.config} not found, check model spec or config_path")
raise ValueError(
f"Config {self.config} not found, check model spec or config_path"
)
if not os.path.exists(self.ckpt):
raise ValueError(f"Checkpoint {self.ckpt} not found, check model spec or config_path")
raise ValueError(
f"Checkpoint {self.ckpt} not found, check model spec or config_path"
)
self.device = device
self.model = self._load_model(device=device, use_fp16=use_fp16)