From e361675f2ea8bc73cbcc5c47eb27838220aa218c Mon Sep 17 00:00:00 2001
From: Vladimir Mandic <mandic00@live.com>
Date: Wed, 1 Jul 2026 14:09:08 +0200
Subject: [PATCH] force safetensors

Signed-off-by: Vladimir Mandic <mandic00@live.com>
---
 pipelines/generic_text_encoder.py | 17 ++++++++++++++++-
 pipelines/generic_transformer.py  | 17 ++++++++++++++++-
 2 files changed, 32 insertions(+), 2 deletions(-)

diff --git a/pipelines/generic_text_encoder.py b/pipelines/generic_text_encoder.py
index 8da8039d5..11cd587eb 100644
--- a/pipelines/generic_text_encoder.py
+++ b/pipelines/generic_text_encoder.py
@@ -49,7 +49,20 @@ def load_local_file(local_file, cls_name, quant_type): # t5-only
     return text_encoder
 
 
-def load_text_encoder(repo_id, cls_name, load_config=None, subfolder="text_encoder", allow_quant=True, allow_shared=True, variant=None, dtype=None, modules_to_not_convert=None, modules_dtype_dict=None, **kwargs):
+def load_text_encoder(
+        repo_id,
+        cls_name,
+        load_config=None,
+        subfolder="text_encoder",
+        allow_quant=True,
+        allow_shared=True,
+        variant=None,
+        dtype=None,
+        modules_to_not_convert=None,
+        modules_dtype_dict=None,
+        use_safetensors=True,
+        **kwargs):
+
     if shared.state.interrupted:
         return None
     if repo_id is None or repo_id.lower() == 'none':
@@ -69,6 +82,8 @@ def load_text_encoder(repo_id, cls_name, load_config=None, subfolder="text_encod
         load_args.pop('torch_dtype', None)
         dtype = dtype or devices.dtype
         load_args['dtype'] = dtype
+        if use_safetensors:
+            load_args['use_safetensors'] = True
 
         # 1. load override from local file
         if (shared.opts.sd_text_encoder is not None) and (shared.opts.sd_text_encoder != 'Default') and (text_encoder is None):
diff --git a/pipelines/generic_transformer.py b/pipelines/generic_transformer.py
index 87ec7741b..142b5ce45 100644
--- a/pipelines/generic_transformer.py
+++ b/pipelines/generic_transformer.py
@@ -9,7 +9,20 @@ from pipelines.generic_map import transformers_map
 debug = os.environ.get('SD_LOAD_DEBUG', None) is not None
 
 
-def load_transformer(repo_id, cls_name, load_config=None, subfolder="transformer", allow_quant=True, variant=None, dtype=None, modules_to_not_convert=None, modules_dtype_dict=None, native_spec=None, **kwargs):
+def load_transformer(
+        repo_id,
+        cls_name,
+        load_config=None,
+        subfolder="transformer",
+        allow_quant=True,
+        variant=None,
+        dtype=None,
+        modules_to_not_convert=None,
+        modules_dtype_dict=None,
+        use_safetensors=True,
+        native_spec=None,
+        **kwargs):
+
     """Load a DiT transformer from the base repo, or from a user-selected
     single file when the UNET dropdown (``shared.opts.sd_unet``) is set.
 
@@ -45,6 +58,8 @@ def load_transformer(repo_id, cls_name, load_config=None, subfolder="transformer
                 load_args['subfolder'] = subfolder
             if variant is not None:
                 load_args['variant'] = variant
+            if use_safetensors:
+                load_args['use_safetensors'] = True
             return cls_name.from_pretrained(
                 repo_id,
                 cache_dir=shared.opts.hfcache_dir,