Update feifeilib/feifeimodload.py

feifeilib/feifeimodload.py

@@ -14,10 +14,15 @@ def feifeimodload():
     taef1 = AutoencoderTiny.from_pretrained("aifeifei798/taef1", torch_dtype=dtype).to(
         device
     )
-    pipe = DiffusionPipeline.from_pretrained(
-        "aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype, vae=taef1
-    ).to(device)
+    
+    # pipe = DiffusionPipeline.from_pretrained(
+    #     "aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype, vae=taef1
+    # ).to(device)
 
+    pipe = DiffusionPipeline.from_pretrained(
+        "shuttleai/shuttle-3.1-aesthetic", torch_dtype=dtype, vae=taef1
+    ).to(device)    
+    
     pipe.load_lora_weights(
         hf_hub_download("aifeifei798/feifei-flux-lora-v1", "feifei.safetensors"),
         adapter_name="feifei",