flux resolution fix (#142)

BitMind-AI · Jan 28, 2025 · 49631c5 · 49631c5
1 parent 0696a60
commit 49631c5
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/bitmind/validator/config.py b/bitmind/validator/config.py
@@ -129,8 +129,7 @@
             "guidance_scale": 2,
             "num_inference_steps": {"min": 50, "max": 125},
             "generator": torch.Generator("cuda" if torch.cuda.is_available() else "cpu"),
-            "height": [512, 768],
-            "width": [512, 768]
+            "resolution": [512, 768]
         },
         "enable_model_cpu_offload": False
     },