Update feifeilib/feifeimodload.py

feifeilib/feifeimodload.py

@@ -11,9 +11,9 @@ 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", torch_dtype=dtype, vae=taef1