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