From f4b27a39af3a0481a33bb2fe7d71fc7c915a1db1 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Tue, 2 Dec 2025 06:35:06 -0800
Subject: [PATCH] Use dtype instead of torch_dtype

---
 modules/image_models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/image_models.py b/modules/image_models.py
index e6f9a172..618ccfb4 100644
--- a/modules/image_models.py
+++ b/modules/image_models.py
@@ -120,7 +120,7 @@ def load_image_model(model_name, dtype='bfloat16', attn_backend='sdpa', cpu_offl
 
         # Load the pipeline
         load_kwargs = {
-            "torch_dtype": target_dtype,
+            "dtype": target_dtype,
             "low_cpu_mem_usage": True,
         }