Update app.py

app.py

@@ -42,7 +42,7 @@ def infer(
 ):  
     ckpt_dir='./lora_pussinboots_logos'
     unet_sub_dir = os.path.join(ckpt_dir, "unet")
-    text_encoder_sub_dir = os.path.join(ckpt_dir, "text_encoder")
+    #text_encoder_sub_dir = os.path.join(ckpt_dir, "text_encoder")
 
     if model_id is None:
         raise ValueError("Please specify the base model name or path")
@@ -100,14 +100,14 @@ def infer(
                                                        safety_checker=None).to(device)
 
     pipe.unet = PeftModel.from_pretrained(pipe.unet, unet_sub_dir)
-    pipe.text_encoder = PeftModel.from_pretrained(pipe.text_encoder, text_encoder_sub_dir)
+    #pipe.text_encoder = PeftModel.from_pretrained(pipe.text_encoder, text_encoder_sub_dir)
 
     pipe.unet.load_state_dict({k: lora_scale*v for k, v in pipe.unet.state_dict().items()})
-    pipe.text_encoder.load_state_dict({k: lora_scale*v for k, v in pipe.text_encoder.state_dict().items()})
+    #pipe.text_encoder.load_state_dict({k: lora_scale*v for k, v in pipe.text_encoder.state_dict().items()})
     
     if torch_dtype in (torch.float16, torch.bfloat16):
         pipe.unet.half()
-        pipe.text_encoder.half()
+        #pipe.text_encoder.half()
 
     if ip_adapter_checkbox:
         pipe.load_ip_adapter("h94/IP-Adapter", subfolder="models", weight_name="ip-adapter-plus_sd15.bin")