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)
|