Преглед на файлове

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

Timothy J. Baek преди 9 месеца
родител
ревизия
e1e69cfbcb
променени са 1 файла, в които са добавени 2 реда и са изтрити 2 реда
  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"