JackAILab hysts HF staff commited on
Commit
5f9444e
1 Parent(s): 5d363d3

- Fix (c72745591176483fbd3a9bfaf32a5386e25e97ac)


Co-authored-by: hysts <hysts@users.noreply.huggingface.co>

README.md CHANGED
@@ -4,7 +4,7 @@ emoji: 🔥
4
  colorFrom: yellow
5
  colorTo: yellow
6
  sdk: gradio
7
- sdk_version: 4.28.3
8
  app_file: app.py
9
  pinned: false
10
  license: apache-2.0
 
4
  colorFrom: yellow
5
  colorTo: yellow
6
  sdk: gradio
7
+ sdk_version: 4.37.2
8
  app_file: app.py
9
  pinned: false
10
  license: apache-2.0
pipline_StableDiffusion_ConsistentID.py CHANGED
@@ -76,7 +76,7 @@ class ConsistentIDStableDiffusionPipeline(StableDiffusionPipeline):
76
  self.crop_size = 512
77
 
78
  # FaceID
79
- self.app = FaceAnalysis(name="buffalo_l", providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
80
  self.app.prepare(ctx_id=0, det_size=(640, 640))
81
 
82
  ### BiSeNet
 
76
  self.crop_size = 512
77
 
78
  # FaceID
79
+ self.app = FaceAnalysis(name="buffalo_l", providers=['CPUExecutionProvider'])
80
  self.app.prepare(ctx_id=0, det_size=(640, 640))
81
 
82
  ### BiSeNet
requirements.txt CHANGED
@@ -6,9 +6,10 @@ omegaconf
6
  peft
7
  opencv-python
8
  insightface
9
- diffusers==0.25.1
10
  torch==2.0.0
11
  torchvision==0.15.1
12
- transformers
13
  spaces
14
- huggingface-hub
 
 
6
  peft
7
  opencv-python
8
  insightface
9
+ diffusers==0.23.0
10
  torch==2.0.0
11
  torchvision==0.15.1
12
+ transformers==4.31.0
13
  spaces
14
+ huggingface-hub
15
+ sentencepiece