Update app.py
Browse files
app.py
CHANGED
|
@@ -63,8 +63,8 @@ from open_clip import tokenizer
|
|
| 63 |
import open_clip
|
| 64 |
|
| 65 |
config = OmegaConf.load("latent-diffusion/configs/latent-diffusion/txt2img-1p4B-eval.yaml")
|
| 66 |
-
|
| 67 |
-
pl_sd = torch.load(
|
| 68 |
sd = pl_sd["state_dict"]
|
| 69 |
model = instantiate_from_config(config.model)
|
| 70 |
m, u = model.load_state_dict(sd, strict=False)
|
|
|
|
| 63 |
import open_clip
|
| 64 |
|
| 65 |
config = OmegaConf.load("latent-diffusion/configs/latent-diffusion/txt2img-1p4B-eval.yaml")
|
| 66 |
+
|
| 67 |
+
pl_sd = torch.load(model_path_e, map_location="cuda")
|
| 68 |
sd = pl_sd["state_dict"]
|
| 69 |
model = instantiate_from_config(config.model)
|
| 70 |
m, u = model.load_state_dict(sd, strict=False)
|