Spaces:
Running
on
Zero
Running
on
Zero
Update app.py (#2)
Browse files- Update app.py (e9b6ecf86d6c37cae42b2c85df377818bd8839e6)
app.py
CHANGED
|
@@ -39,7 +39,7 @@ 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.load_state_dict(torch.load(bise_net_cp_path, map_location=
|
| 43 |
|
| 44 |
### Load consistentID_model checkpoint
|
| 45 |
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=device))
|
| 43 |
|
| 44 |
### Load consistentID_model checkpoint
|
| 45 |
pipe.load_ConsistentID_model(
|