Tiny changes

app.py

@@ -12,7 +12,7 @@ from lib.pipline_ConsistentID import ConsistentIDPipeline
 import argparse
 parser = argparse.ArgumentParser()
 parser.add_argument('--base_model_path', type=str, 
-                    default="models/SG161222/Realistic_Vision_V4.0_noVAE")
+                    default="models/Realistic_Vision_V4.0_noVAE")
 parser.add_argument('--gpu', type=int, default=0)
 args = parser.parse_args()
 
@@ -22,8 +22,6 @@ device = f"cuda:{args.gpu}"
 pipe = ConsistentIDPipeline.from_pretrained(
     args.base_model_path, 
     torch_dtype=torch.float16, 
-    safety_checker=None, # use_safetensors=True, 
-    variant="fp16"
 ).to(device)
 
 ### Load consistentID_model checkpoint

models/Realistic_Vision_V4.0_noVAE

@@ -0,0 +1 @@
+/home/lish/adaprompt/models/Realistic_Vision_V4.0_noVAE

models/Realistic_Vision_V6.0_B1_noVAE

@@ -0,0 +1 @@
+/home/lish/adaprompt/models/Realistic_Vision_V6.0_B1_noVAE

requirements.txt

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