Fix (#9)

- Fix (c72745591176483fbd3a9bfaf32a5386e25e97ac)


Co-authored-by: hysts <[email protected]>

README.md

@@ -4,7 +4,7 @@ emoji: 🔥
 colorFrom: yellow
 colorTo: yellow
 sdk: gradio
-sdk_version: 4.28.3
+sdk_version: 4.37.2
 app_file: app.py
 pinned: false
 license: apache-2.0

pipline_StableDiffusion_ConsistentID.py

@@ -76,7 +76,7 @@ class ConsistentIDStableDiffusionPipeline(StableDiffusionPipeline):
         self.crop_size = 512
 
         # FaceID
-        self.app = FaceAnalysis(name="buffalo_l", providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
+        self.app = FaceAnalysis(name="buffalo_l", providers=['CPUExecutionProvider'])
         self.app.prepare(ctx_id=0, det_size=(640, 640))
 
         ### BiSeNet

requirements.txt

@@ -6,9 +6,10 @@ omegaconf
 peft
 opencv-python
 insightface
-diffusers==0.25.1
+diffusers==0.23.0
 torch==2.0.0
 torchvision==0.15.1
-transformers
+transformers==4.31.0
 spaces
-huggingface-hub
+huggingface-hub
+sentencepiece