multimodalart HF Staff commited on
Commit
0669acc
·
verified ·
1 Parent(s): 9bc2183

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- print(f"Loading model from {ckpt}")
67
- pl_sd = torch.load(ckpt, 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)
 
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)