Update feifeilib/feifeimodload.py

feifeilib/feifeimodload.py

@@ -11,14 +11,18 @@ def feifeimodload():
 
     dtype = torch.bfloat16
     device = "cuda" if torch.cuda.is_available() else "cpu"
-    taef1 = AutoencoderTiny.from_pretrained("aifeifei798/taef1", torch_dtype=dtype).to(
-        device
-    )
+    #taef1 = AutoencoderTiny.from_pretrained("aifeifei798/taef1", torch_dtype=dtype).to(
+    #    device
+    #)
     
+    #pipe = DiffusionPipeline.from_pretrained(
+    #    "aifeifei798/DarkIdol-flux-v1.1", torch_dtype=dtype, vae=taef1
+    #).to(device)
+
     pipe = DiffusionPipeline.from_pretrained(
-        "aifeifei798/DarkIdol-flux-v1.1", torch_dtype=dtype, vae=taef1
+        "shuttleai/shuttle-3.1-aesthetic", torch_dtype=dtype
     ).to(device)
-
+    
     pipe.load_lora_weights(
         hf_hub_download("aifeifei798/feifei-flux-lora-v1", "feifei.safetensors"),
         adapter_name="feifei",