Spaces:

Purple11
/

Grounded-Diffusion

Runtime error

Purple11 commited on Feb 18, 2023

Commit

b75e089

1 Parent(s): ed31fc4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -47,7 +47,7 @@ def numpy_to_pil(images):
 def load_model_from_config(config, ckpt, verbose=False):
-    # print(f"Loading model from {ckpt}")
     pl_sd = torch.load(ckpt, map_location="cpu")
     sd = pl_sd["state_dict"]
     model = instantiate_from_config(config.model)

 def load_model_from_config(config, ckpt, verbose=False):
+    print(f"Loading model from {ckpt}")
     pl_sd = torch.load(ckpt, map_location="cpu")
     sd = pl_sd["state_dict"]
     model = instantiate_from_config(config.model)