change sd1.5 to modelscope

app.py

@@ -31,6 +31,7 @@ else:
 
 # download models
 from huggingface_hub import hf_hub_download
+from modelscope import snapshot_download
 
 hf_hub_download(repo_id="FoivosPar/Arc2Face", filename="arc2face/config.json", local_dir="./models")
 hf_hub_download(repo_id="FoivosPar/Arc2Face", filename="arc2face/diffusion_pytorch_model.safetensors", local_dir="./models")
@@ -38,12 +39,14 @@ hf_hub_download(repo_id="FoivosPar/Arc2Face", filename="encoder/config.json", lo
 hf_hub_download(repo_id="FoivosPar/Arc2Face", filename="encoder/pytorch_model.bin", local_dir="./models")
 hf_hub_download(repo_id="FoivosPar/Arc2Face", filename="arcface.onnx", local_dir="./models/antelopev2")
 
+base_model = snapshot_download('AI-ModelScope/stable-diffusion-v1-5', cache_dir='./models')
+
 # Load face detection and recognition package
 app = FaceAnalysis(name='antelopev2', root='./', providers=['CPUExecutionProvider'])
 app.prepare(ctx_id=0, det_size=(640, 640))
 
 # Load pipeline
-base_model = 'runwayml/stable-diffusion-v1-5'
+#base_model = 'runwayml/stable-diffusion-v1-5'
 encoder = CLIPTextModelWrapper.from_pretrained(
     'models', subfolder="encoder", torch_dtype=dtype
 )

requirements.txt

@@ -5,6 +5,7 @@ diffusers==0.23.0
 transformers==4.34.1
 peft
 accelerate
+modelscope
 insightface
 onnxruntime-gpu
 gradio