From ad34f4329189606f554c1256324dc4d69df25995 Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Tue, 13 Aug 2024 11:07:45 -0700
Subject: [PATCH] Fix new paths to flux models

---
 comfy/model_downloader.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/comfy/model_downloader.py b/comfy/model_downloader.py
index 9441cad61..f5778dccc 100644
--- a/comfy/model_downloader.py
+++ b/comfy/model_downloader.py
@@ -402,7 +402,7 @@ KNOWN_APPROX_VAES: Final[KnownDownloadables] = KnownDownloadables([
 KNOWN_VAES: Final[KnownDownloadables] = KnownDownloadables([
     HuggingFile("stabilityai/sdxl-vae", "sdxl_vae.safetensors"),
     HuggingFile("stabilityai/sd-vae-ft-mse-original", "vae-ft-mse-840000-ema-pruned.safetensors"),
-    HuggingFile("black-forest-labs/FLUX.1-schnell", "ae.sft"),
+    HuggingFile("black-forest-labs/FLUX.1-schnell", "ae.safetensors"),
 ], folder_name="vae")
 
 KNOWN_HUGGINGFACE_MODEL_REPOS: Final[Set[str]] = {
@@ -414,8 +414,8 @@ KNOWN_HUGGINGFACE_MODEL_REPOS: Final[Set[str]] = {
 
 KNOWN_UNET_MODELS: Final[KnownDownloadables] = KnownDownloadables([
     HuggingFile("ByteDance/Hyper-SD", "Hyper-SDXL-1step-Unet-Comfyui.fp16.safetensors"),
-    HuggingFile("black-forest-labs/FLUX.1-schnell", "flux1-schnell.sft"),
-    HuggingFile("black-forest-labs/FLUX.1-dev", "flux1-dev.sft"),
+    HuggingFile("black-forest-labs/FLUX.1-schnell", "flux1-schnell.safetensors"),
+    HuggingFile("black-forest-labs/FLUX.1-dev", "flux1-dev.safetensors"),
 ], folder_name="unet")
 
 KNOWN_CLIP_MODELS: Final[KnownDownloadables] = KnownDownloadables([