Update app.py

app.py

@@ -41,7 +41,10 @@ device = "cuda"
 script_directory = os.path.dirname(os.path.realpath(__file__))
 
 # download ConsistentID checkpoint to cache
-base_model_path = "SG161222/RealVisXL_V3.0"
+#base_model_path = "SG161222/RealVisXL_V3.0"
+#base_model_path = "SG161222/RealVisXL_V4.0"
+base_model_path = "SG161222/RealVisXL_V5.0"
+
 consistentID_path = hf_hub_download(repo_id="JackAILab/ConsistentID", filename="ConsistentID_SDXL-v1.bin", repo_type="model")
 
 ### Load base model
@@ -49,7 +52,7 @@ pipe = ConsistentIDStableDiffusionXLPipeline.from_pretrained(
     base_model_path, 
     torch_dtype=torch.float16, 
     safety_checker=None, # use_safetensors=True, 
-    variant="fp16"
+    #variant="fp16"
 ).to(device)
 
 ### Load other pretrained models