fix: backend format test

204a4fbe · Michael Poluektov · 54981762 · 204a4fbe
Commit 204a4fbe authored Aug 08, 2024 by Michael Poluektov
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

backend/apps/images/utils/comfyui.py backend/apps/images/utils/comfyui.py +3 -3

No files found.
--- a/backend/apps/images/utils/comfyui.py
+++ b/backend/apps/images/utils/comfyui.py
@@ -377,9 +377,9 @@ async def comfyui_generate_image(
            comfyui_prompt["12"]["inputs"]["weight_dtype"] = payload.flux_weight_dtype

        if payload.flux_fp8_clip:
-            comfyui_prompt["11"]["inputs"]["clip_name2"] = (
-                "t5xxl_fp8_e4m3fn.safetensors"
-            )
+            comfyui_prompt["11"]["inputs"][
+                "clip_name2"
+            ] = "t5xxl_fp8_e4m3fn.safetensors"

    comfyui_prompt["5"]["inputs"]["batch_size"] = payload.n
    comfyui_prompt["5"]["inputs"]["width"] = payload.width