Spaces:

aifeifei798
/

allinone-DarkIdol-flux

Running on Zero

aifeifei798 commited on Dec 15, 2024

Commit

bf71fb2

verified ·

1 Parent(s): 740e658

Update feifeilib/feifeimodload.py

Files changed (1) hide show

feifeilib/feifeimodload.py CHANGED Viewed

@@ -20,7 +20,7 @@ def feifeimodload():
       torch_dtype=torch.bfloat16
     )
-    pipe = DiffusionPipeline.from_pretrained(
         "aifeifei798/DarkIdol-flux-v1", controlnet=controlnet, torch_dtype=dtype,
     ).to(device)

       torch_dtype=torch.bfloat16
     )
+    pipe = FluxControlNetPipeline.from_pretrained(
         "aifeifei798/DarkIdol-flux-v1", controlnet=controlnet, torch_dtype=dtype,
     ).to(device)