Spaces:

JackAILab
/

ConsistentID

Running on Zero

JackAILab commited on May 8

Commit

272b583

•

1 Parent(s): 0f4bb3b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -39,7 +39,8 @@ 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
 ### Load consistentID_model checkpoint
 pipe.load_ConsistentID_model(

 ## 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.cuda()
+bise_net.load_state_dict(torch.load(bise_net_cp_path)) # device fail
 ### Load consistentID_model checkpoint
 pipe.load_ConsistentID_model(