Timothy J. Baek 9 miesięcy temu
rodzic
commit
8062866973

+ 1 - 0
backend/apps/images/main.py

@@ -92,6 +92,7 @@ app.state.config.COMFYUI_FLUX = COMFYUI_FLUX
 app.state.config.COMFYUI_FLUX_WEIGHT_DTYPE = COMFYUI_FLUX_WEIGHT_DTYPE
 app.state.config.COMFYUI_FLUX_FP8_CLIP = COMFYUI_FLUX_FP8_CLIP
 
+
 def get_automatic1111_api_auth():
     if app.state.config.AUTOMATIC1111_API_AUTH == None:
         return ""

+ 4 - 2
backend/apps/images/utils/comfyui.py

@@ -375,9 +375,11 @@ def comfyui_generate_image(
 
         if payload.flux_weight_dtype:
             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