Update feifeilib/feifeimodload.py

feifeilib/feifeimodload.py

@@ -2,7 +2,8 @@ import torch
 import spaces
 from diffusers import (
     DiffusionPipeline,
-    AutoencoderTiny
+    AutoencoderTiny,
+    FluxImg2ImgPipeline
 )
 from huggingface_hub import hf_hub_download
 
@@ -39,5 +40,8 @@ def feifeimodload():
     pipe.vae.enable_slicing()
     pipe.vae.enable_tiling()
     # pipe.unload_lora_weights()
+    imgpipe = FluxImg2ImgPipeline.from_pretrained(
+        "aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype
+    ).to(device)
     torch.cuda.empty_cache()
-    return pipe
+    return pipe, imgpipe