Spaces:

aifeifei798
/

allinone-DarkIdol-flux

Running on Zero

aifeifei798 commited on Dec 13, 2024

Commit

dfd2cf6

verified ·

1 Parent(s): 140c980

Update feifeilib/feifeimodload.py

Files changed (1) hide show

feifeilib/feifeimodload.py CHANGED Viewed

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

     device = "cuda" if torch.cuda.is_available() else "cpu"
     pipe = DiffusionPipeline.from_pretrained(
+        "aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype
     ).to(device)
     pipe.load_lora_weights(