Update app.py

app.py

@@ -16,7 +16,7 @@ from huggingface_hub import hf_hub_download
 from diffusers import DiffusionPipeline, AutoencoderTiny
 import safetensors.torch
 from safetensors.torch import load_file
-from custom_pipeline import FluxWithCFGPipeline
+from pipeline import FluxWithCFGPipeline
 from transformers import CLIPModel, CLIPProcessor, CLIPConfig
 import gc
 
@@ -30,8 +30,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 torch.backends.cuda.matmul.allow_tf32 = True
 
 dtype = torch.bfloat16
-pipe = FluxWithCFGPipeline.from_pretrained(
-    "ostris/OpenFLUX.1", custom_pipeline="AlekseyCalvin/OpenFlux_Lorasoonr", torch_dtype=dtype
+pipe = FluxWithCFGPipeline.from_pretrained("ostris/OpenFLUX.1", torch_dtype=dtype
 ).to("cuda")
 pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype).to("cuda")