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

e1e69cfb · Timothy J. Baek · 30c44d43 · e1e69cfb
Commit e1e69cfb authored Aug 13, 2024 by Timothy J. Baek
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

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

No files found.
--- a/backend/apps/images/utils/comfyui.py
+++ b/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"