Update app.py

app.py

@@ -35,9 +35,9 @@ if mps_available:
     torch_device = "cpu"
     torch_dtype = torch.float32
 
-# pipe = StableDiffusionPipeline.from_single_file(
-#     "https://huggingface.co/wanghuging/skin_demo/blob/main/skin_demo.safetensors"
-# )
+t2i_pipe = StableDiffusionPipeline.from_single_file(
+    "https://huggingface.co/wanghuging/skin_demo/blob/main/skin_demo.safetensors"
+)
     
 if SAFETY_CHECKER == "True":
     i2i_pipe = AutoPipelineForImage2Image.from_pretrained(
@@ -45,14 +45,14 @@ if SAFETY_CHECKER == "True":
         torch_dtype=torch_dtype,
         variant="fp16" if torch_dtype == torch.float16 else "fp32",
     )
-    t2i_pipe = AutoPipelineForText2Image.from_pretrained(
-        #"stabilityai/sdxl-turbo",
-        # "wanghuging/demo_model",
-        #"stabilityai/stable-diffusion-xl-base-1.0",
-        "stabilityai/stable-diffusion-2-1",
-        torch_dtype=torch_dtype,
-        variant="fp16" #if torch_dtype == torch.float16 else "fp32",
-    )
+    # t2i_pipe = AutoPipelineForText2Image.from_pretrained(
+    #     #"stabilityai/sdxl-turbo",
+    #     # "wanghuging/demo_model",
+    #     #"stabilityai/stable-diffusion-xl-base-1.0",
+    #     "stabilityai/stable-diffusion-2-1",
+    #     torch_dtype=torch_dtype,
+    #     variant="fp16" #if torch_dtype == torch.float16 else "fp32",
+    # )
 else:
     i2i_pipe = AutoPipelineForImage2Image.from_pretrained(
         "stabilityai/sdxl-turbo",
@@ -60,17 +60,17 @@ else:
         torch_dtype=torch_dtype,
         variant="fp16" if torch_dtype == torch.float16 else "fp32",
     )
-    t2i_pipe = AutoPipelineForText2Image.from_pretrained(
-        #"stabilityai/sdxl-turbo",
-        # "wanghuging/demo_model",
-        # "stabilityai/stable-diffusion-xl-base-1.0",      
-        "stabilityai/stable-diffusion-2-1",        
-        safety_checker=None,
-        torch_dtype=torch_dtype,
-        variant="fp16" #if torch_dtype == torch.float16 else "fp32",
-    )
-
-t2i_pipe.load_lora_weights("wanghuging/skin_demo", weight_name="skin_demo.safetensors")
+    # t2i_pipe = AutoPipelineForText2Image.from_pretrained(
+    #     #"stabilityai/sdxl-turbo",
+    #     # "wanghuging/demo_model",
+    #     # "stabilityai/stable-diffusion-xl-base-1.0",      
+    #     "stabilityai/stable-diffusion-2-1",        
+    #     safety_checker=None,
+    #     torch_dtype=torch_dtype,
+    #     variant="fp16" #if torch_dtype == torch.float16 else "fp32",
+    # )
+
+# t2i_pipe.load_lora_weights("wanghuging/skin_demo", weight_name="skin_demo.safetensors")
 t2i_pipe.to(device=torch_device, dtype=torch_dtype).to(device)
 t2i_pipe.set_progress_bar_config(disable=True)
 i2i_pipe.to(device=torch_device, dtype=torch_dtype).to(device)