1

app.py

@@ -29,7 +29,7 @@ import open_clip
 
 def load_model_from_config(config, ckpt, verbose=False):
     print(f"Loading model from {ckpt}")
-    pl_sd = torch.load(ckpt, map_location="cuda")
+    #pl_sd = torch.load(ckpt, map_location="cuda")
     #please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.
     pl_sd = torch.load(ckpt, map_location=torch.device('cpu'))
     sd = pl_sd["state_dict"]