From 93397949d2b37370c37796e9edbe24d33227d087 Mon Sep 17 00:00:00 2001
From: "Zak B. Elep" <zakame@zakame.net>
Date: Tue, 15 Oct 2024 23:55:28 +0800
Subject: [PATCH] fix: typo on flux model preloading

Still referring to the sd version of the pipeline class.
---
 flux/txt2image.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/flux/txt2image.py b/flux/txt2image.py
index bf2f7294..5ebec81a 100644
--- a/flux/txt2image.py
+++ b/flux/txt2image.py
@@ -77,7 +77,7 @@ if __name__ == "__main__":
         nn.quantize(flux.clip, class_predicate=quantization_predicate)
 
     if args.preload_models:
-        sd.ensure_models_are_loaded()
+        flux.ensure_models_are_loaded()
 
     # Make the generator
     latent_size = to_latent_size(args.image_size)