Update app.py

app.py

@@ -9,7 +9,7 @@ import logging
 from diffusers import DiffusionPipeline, AutoencoderTiny, AutoencoderKL, AutoPipelineForImage2Image
 from huggingface_hub import login
 from diffusers.utils import load_image
-from lora_loading_patch import load_lora_into_transformer
+#from lora_loading_patch import load_lora_into_transformer
 import time
 from datetime import datetime
 from io import BytesIO
@@ -22,6 +22,8 @@ import re
 import json
 import random
 import string
+from diffusers import FluxPipeline
+
 
 # Login Hugging Face Hub
 HF_TOKEN = os.environ.get("HF_TOKEN")
@@ -37,10 +39,10 @@ print(device)
 base_model = "black-forest-labs/FLUX.1-Krea-dev"
 # load pipe
 
-txt2img_pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype)
+txt2img_pipe = FluxPipeline.from_pretrained(base_model, torch_dtype=dtype)
 
 txt2img_pipe = txt2img_pipe.to(device)
-txt2img_pipe.__class__.load_lora_into_transformer = classmethod(load_lora_into_transformer)
+#txt2img_pipe.__class__.load_lora_into_transformer = classmethod(load_lora_into_transformer)
 
 MAX_SEED = 2**32 - 1