Update feifeilib/feifeimodload.py

feifeilib/feifeimodload.py

@@ -14,13 +14,13 @@ def feifeimodload():
     taef1 = AutoencoderTiny.from_pretrained("aifeifei798/taef1", torch_dtype=dtype).to(
         device
     )
-    # pipe = DiffusionPipeline.from_pretrained(
-    #     "sayakpaul/FLUX.1-merged", torch_dtype=dtype, vae=taef1
-    # ).to(device)
-
     pipe = DiffusionPipeline.from_pretrained(
-        "aifeifei798/shuttle-3-diffusion", torch_dtype=dtype, vae=taef1
+        "sayakpaul/FLUX.1-merged", torch_dtype=dtype, vae=taef1
     ).to(device)
+
+    #pipe = DiffusionPipeline.from_pretrained(
+    #    "aifeifei798/shuttle-3-diffusion", torch_dtype=dtype, vae=taef1
+    #).to(device)
     
     # pipe = DiffusionPipeline.from_pretrained(
     #     "black-forest-labs/FLUX.1-schnell", torch_dtype=dtype, vae=taef1