fix demo

app.py

@@ -17,7 +17,7 @@ from insightface.app import FaceAnalysis
 from PIL import Image
 import numpy as np
 import random
-import os
+#import os
 
 import gradio as gr
 
@@ -34,7 +34,7 @@ else:
 # download models
 from huggingface_hub import hf_hub_download
 #from modelscope import snapshot_download
-from modelscope.hub.file_download import model_file_download
+#from modelscope.hub.file_download import model_file_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")
@@ -43,15 +43,15 @@ hf_hub_download(repo_id="FoivosPar/Arc2Face", filename="encoder/pytorch_model.bi
 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')
-model_dir = model_file_download(model_id='AI-ModelScope/stable-diffusion-v1-5', file_path='model_index.json', cache_dir='./models')
-base_model = os.path.dirname(model_dir)
-_ = model_file_download(model_id='AI-ModelScope/stable-diffusion-v1-5', file_path='scheduler/scheduler_config.json', cache_dir='./models')
-_ = model_file_download(model_id='AI-ModelScope/stable-diffusion-v1-5', file_path='tokenizer/merges.txt', cache_dir='./models')
-_ = model_file_download(model_id='AI-ModelScope/stable-diffusion-v1-5', file_path='tokenizer/special_tokens_map.json', cache_dir='./models')
-_ = model_file_download(model_id='AI-ModelScope/stable-diffusion-v1-5', file_path='tokenizer/tokenizer_config.json', cache_dir='./models')
-_ = model_file_download(model_id='AI-ModelScope/stable-diffusion-v1-5', file_path='tokenizer/vocab.json', cache_dir='./models')
-_ = model_file_download(model_id='AI-ModelScope/stable-diffusion-v1-5', file_path='vae/config.json', cache_dir='./models')
-_ = model_file_download(model_id='AI-ModelScope/stable-diffusion-v1-5', file_path='vae/diffusion_pytorch_model.safetensors', cache_dir='./models')
+#model_dir = model_file_download(model_id='AI-ModelScope/stable-diffusion-v1-5', file_path='model_index.json', cache_dir='./models')
+#base_model = os.path.dirname(model_dir)
+#_ = model_file_download(model_id='AI-ModelScope/stable-diffusion-v1-5', file_path='scheduler/scheduler_config.json', cache_dir='./models')
+#_ = model_file_download(model_id='AI-ModelScope/stable-diffusion-v1-5', file_path='tokenizer/merges.txt', cache_dir='./models')
+#_ = model_file_download(model_id='AI-ModelScope/stable-diffusion-v1-5', file_path='tokenizer/special_tokens_map.json', cache_dir='./models')
+#_ = model_file_download(model_id='AI-ModelScope/stable-diffusion-v1-5', file_path='tokenizer/tokenizer_config.json', cache_dir='./models')
+#_ = model_file_download(model_id='AI-ModelScope/stable-diffusion-v1-5', file_path='tokenizer/vocab.json', cache_dir='./models')
+#_ = model_file_download(model_id='AI-ModelScope/stable-diffusion-v1-5', file_path='vae/config.json', cache_dir='./models')
+#_ = model_file_download(model_id='AI-ModelScope/stable-diffusion-v1-5', file_path='vae/diffusion_pytorch_model.safetensors', cache_dir='./models')
 
 # Load face detection and recognition package
 app = FaceAnalysis(name='antelopev2', root='./', providers=['CPUExecutionProvider'])
@@ -59,6 +59,7 @@ app.prepare(ctx_id=0, det_size=(640, 640))
 
 # Load pipeline
 #base_model = 'runwayml/stable-diffusion-v1-5'
+base_model = 'ashllay/stable-diffusion-v1-5-archive'
 encoder = CLIPTextModelWrapper.from_pretrained(
     'models', subfolder="encoder", torch_dtype=dtype
 )
@@ -71,7 +72,6 @@ pipeline = StableDiffusionPipeline.from_pretrained(
         unet=unet,
         torch_dtype=dtype,
         safety_checker=None,
-        feature_extractor=None
     )
 pipeline.scheduler = DPMSolverMultistepScheduler.from_config(pipeline.scheduler.config)
 pipeline = pipeline.to(device)

requirements.txt

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