Update app.py

app.py

@@ -11,17 +11,17 @@ def feifeimodload():
     dtype = torch.bfloat16
     device = "cuda" if torch.cuda.is_available() else "cpu"
     
-    #pipe = DiffusionPipeline.from_pretrained(
-    #    "aifeifei798/DarkIdol-flux-v1.1", torch_dtype=dtype
-    #).to(device)
-
     pipe = DiffusionPipeline.from_pretrained(
-        "city96/Flux.1-Heavy-17B", torch_dtype=dtype
+        "aifeifei798/DarkIdol-flux-v1.1", torch_dtype=dtype
     ).to(device)
+
+    #pipe = DiffusionPipeline.from_pretrained(
+    #    "city96/Flux.1-Heavy-17B", torch_dtype=dtype
+    #).to(device)
     
     pipe.vae.enable_slicing()
     pipe.vae.enable_tiling()
-    pipe.unload_lora_weights()
+    #pipe.unload_lora_weights()
     torch.cuda.empty_cache()
     return pipe