Explorar o código

refac: `sft` -> default to `safetensors`

Timothy J. Baek hai 9 meses
pai
achega
e1e69cfbcb
Modificáronse 1 ficheiros con 2 adicións e 2 borrados
  1. 2 2
      backend/apps/images/utils/comfyui.py

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

@@ -170,7 +170,7 @@ FLUX_DEFAULT_PROMPT = """
     },
     "10": {
         "inputs": {
-            "vae_name": "ae.sft"
+            "vae_name": "ae.safetensors"
         },
         "class_type": "VAELoader"
     },
@@ -184,7 +184,7 @@ FLUX_DEFAULT_PROMPT = """
     },
     "12": {
         "inputs": {
-            "unet_name": "flux1-dev.sft",
+            "unet_name": "flux1-dev.safetensors",
             "weight_dtype": "default"
         },
         "class_type": "UNETLoader"