Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -39,8 +39,8 @@ pipe = ConsistentIDStableDiffusionPipeline.from_pretrained(
|
|
| 39 |
## BiSenet
|
| 40 |
bise_net_cp_path = hf_hub_download(repo_id="JackAILab/ConsistentID", filename="face_parsing.pth", local_dir="./checkpoints")
|
| 41 |
bise_net = BiSeNet(n_classes = 19)
|
|
|
|
| 42 |
bise_net.cuda()
|
| 43 |
-
bise_net.load_state_dict(torch.load(bise_net_cp_path)) # device fail
|
| 44 |
|
| 45 |
### Load consistentID_model checkpoint
|
| 46 |
pipe.load_ConsistentID_model(
|
|
|
|
| 39 |
## BiSenet
|
| 40 |
bise_net_cp_path = hf_hub_download(repo_id="JackAILab/ConsistentID", filename="face_parsing.pth", local_dir="./checkpoints")
|
| 41 |
bise_net = BiSeNet(n_classes = 19)
|
| 42 |
+
bise_net.load_state_dict(torch.load(bise_net_cp_path), map_location="cpu") # device fail
|
| 43 |
bise_net.cuda()
|
|
|
|
| 44 |
|
| 45 |
### Load consistentID_model checkpoint
|
| 46 |
pipe.load_ConsistentID_model(
|