Spaces:

XavierJiezou
/

face-mogle

Runtime error

XavierJiezou commited on Apr 12

Commit

41c83f6

verified ·

1 Parent(s): ec50c67

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -703,7 +703,7 @@ class ImageGenerator:
             pipe = "black-forest-labs/FLUX.1-schnell"
             lora_ckpt = "weights"
             moe_ckpt = "weights/mogle.pt"
-            pretrained_ckpt = "black-forest-labs/FLUX.1-dev"
             device = "cuda" if torch.cuda.is_available() else "cpu"
             size = 512
             seed = 42

             pipe = "black-forest-labs/FLUX.1-schnell"
             lora_ckpt = "weights"
             moe_ckpt = "weights/mogle.pt"
+            pretrained_ckpt = "black-forest-labs/FLUX.1-schnell"
             device = "cuda" if torch.cuda.is_available() else "cpu"
             size = 512
             seed = 42