Spaces:

JackAILab
/

ConsistentID

Running on Zero

JackAILab commited on May 8

Commit

50e55b2

•

1 Parent(s): d6a6e70

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -39,7 +39,7 @@ pipe = ConsistentIDStableDiffusionPipeline.from_pretrained(
 ## BiSenet
 bise_net_cp_path = hf_hub_download(repo_id="JackAILab/ConsistentID", filename="face_parsing.pth", local_dir="./checkpoints")
 bise_net = BiSeNet(n_classes = 19)
-bise_net.load_state_dict(torch.load(bise_net_cp_path), map_location="cpu") # device fail
 bise_net.cuda()
 ### Load consistentID_model checkpoint

 ## BiSenet
 bise_net_cp_path = hf_hub_download(repo_id="JackAILab/ConsistentID", filename="face_parsing.pth", local_dir="./checkpoints")
 bise_net = BiSeNet(n_classes = 19)
+bise_net.load_state_dict(torch.load(bise_net_cp_path, map_location="cpu")) # device fail
 bise_net.cuda()
 ### Load consistentID_model checkpoint