Update feifeilib/feifeimodload.py

feifeilib/feifeimodload.py

@@ -1,7 +1,7 @@
 import torch
 import spaces
 from diffusers import (
-    DiffusionPipeline,
+    FluxImg2ImgPipeline,
     AutoencoderTiny,
     FluxImg2ImgPipeline
 )
@@ -13,7 +13,7 @@ def feifeimodload():
     device = "cuda" if torch.cuda.is_available() else "cpu"
 
     
-    pipe = DiffusionPipeline.from_pretrained(
+    pipe = FluxImg2ImgPipeline.from_pretrained(
         "aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype
     ).to(device)