Spaces:

basso4
/

FPT-VTON

Runtime error

basso4 commited on Nov 12, 2024

Commit

d6585d1

verified ·

1 Parent(s): b6e2095

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -48,7 +48,7 @@ if __name__ == "__main__":
     device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
     model = instantiate_from_config(config.model)
-    model.load_state_dict(torch.hub.load_state_dict_from_url("https://huggingface.co/basso4/hitonhd/resolve/main/hitonhd.ckpt")["state_dict"], strict=False)
     model.cuda()
     model.eval()
     model = model.to(device)

     device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
     model = instantiate_from_config(config.model)
+    model.load_state_dict(torch.hub.load_state_dict_from_url("https://huggingface.co/basso4/FPT_DM/resolve/main/fptdm.ckpt")["state_dict"], strict=False)
     model.cuda()
     model.eval()
     model = model.to(device)