Hugging Face's logo

Spaces:
JackAILab
/
ConsistentID
Running on Zero

App Files Community
9

Update app.py

#2
by JackAILab - opened
base: refs/heads/main
←
from: refs/pr/2
Discussion Files changed
+1
-1
Files changed (1) hide show
  1. app.py +1 -1
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=devide))
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(