Update feifeilib/feifeimodload.py

feifeilib/feifeimodload.py

@@ -15,10 +15,14 @@ def feifeimodload():
         device
     )
     
+    #pipe = DiffusionPipeline.from_pretrained(
+    #    "aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype, vae=taef1
+    #).to(device)
+
     pipe = DiffusionPipeline.from_pretrained(
-        "aifeifei798/DarkIdol-flux-v1.1", torch_dtype=dtype, vae=taef1
+        "aifeifei798/DarkIdol-flux-v1.1", torch_dtype=dtype
     ).to(device)
-
+    
     #pipe = DiffusionPipeline.from_pretrained(
     #    "shuttleai/shuttle-3.1-aesthetic", torch_dtype=dtype
     #).to(device)
@@ -72,7 +76,7 @@ def feifeimodload():
         lora_scale=1.0,
     )
 
-    #pipe.vae.enable_tiling()
+    pipe.vae.enable_tiling()
     pipe.unload_lora_weights()
     torch.cuda.empty_cache()
     return pipe