1

app.py

@@ -30,6 +30,8 @@ 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")
+    #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"]
     model = instantiate_from_config(config.model)
     m, u = model.load_state_dict(sd, strict=False)