Spaces:
Runtime error
Runtime error
Commit
·
8c61b5b
1
Parent(s):
5b21366
File name match
Browse files
app.py
CHANGED
@@ -42,7 +42,7 @@ def load_model_from_config(config, ckpt, verbose=False):
|
|
42 |
return model
|
43 |
|
44 |
config = OmegaConf.load("latent-diffusion/configs/latent-diffusion/txt2img-1p4B-eval.yaml") # TODO: Optionally download from same location as ckpt and chnage this logic
|
45 |
-
model = load_model_from_config(config, f"
|
46 |
device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
|
47 |
model = model.to(device)
|
48 |
|
|
|
42 |
return model
|
43 |
|
44 |
config = OmegaConf.load("latent-diffusion/configs/latent-diffusion/txt2img-1p4B-eval.yaml") # TODO: Optionally download from same location as ckpt and chnage this logic
|
45 |
+
model = load_model_from_config(config, f"txt2img-f8-large.ckpt") # TODO: check path
|
46 |
device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
|
47 |
model = model.to(device)
|
48 |
|