From 3f830589d122b792997935da2a92346514c5cc3c Mon Sep 17 00:00:00 2001
From: awsr <43862868+awsr@users.noreply.github.com>
Date: Tue, 24 Mar 2026 05:19:51 -0700
Subject: [PATCH] RUF013 updates and typing update

---
 modules/sd_offload.py     | 2 +-
 modules/sd_offload_aux.py | 2 +-
 modules/sd_te_remote.py   | 2 +-
 modules/sd_unet.py        | 8 ++++----
 4 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/modules/sd_offload.py b/modules/sd_offload.py
index e2568f1f5..da478b883 100644
--- a/modules/sd_offload.py
+++ b/modules/sd_offload.py
@@ -467,7 +467,7 @@ def report_model_stats(module_name, module):
         log.error(f'Module stats: name={module_name} {e}')
 
 
-def apply_balanced_offload(sd_model=None, exclude:list[str]=None, force:bool=False, silent:bool=False):
+def apply_balanced_offload(sd_model=None, exclude: list[str] | None = None, force: bool = False, silent: bool = False):
     global offload_hook_instance # pylint: disable=global-statement
     if shared.opts.diffusers_offload_mode != "balanced":
         return sd_model
diff --git a/modules/sd_offload_aux.py b/modules/sd_offload_aux.py
index 8578b8184..26c8c9fa7 100644
--- a/modules/sd_offload_aux.py
+++ b/modules/sd_offload_aux.py
@@ -33,7 +33,7 @@ def deregister_aux(name: str) -> None:
         debug_move(f'Offload: type=aux op=deregister name={name}')
 
 
-def evict_aux(exclude: str = None, reason: str = 'evict') -> None:
+def evict_aux(exclude: str | None = None, reason: str = 'evict') -> None:
     for name, entry in aux_models.items():
         if name == exclude:
             continue
diff --git a/modules/sd_te_remote.py b/modules/sd_te_remote.py
index 36c920762..6a87df927 100644
--- a/modules/sd_te_remote.py
+++ b/modules/sd_te_remote.py
@@ -8,7 +8,7 @@ from modules.logger import log
 
 
 def get_t5_prompt_embeds(
-    prompt: str | list[str] = None,
+    prompt: str | list[str] | None = None,
     num_images_per_prompt: int = 1, # pylint: disable=unused-argument
     max_sequence_length: int = 512, # pylint: disable=unused-argument
     device: torch.device | None = None,
diff --git a/modules/sd_unet.py b/modules/sd_unet.py
index a3ac1c633..6b5d94545 100644
--- a/modules/sd_unet.py
+++ b/modules/sd_unet.py
@@ -35,11 +35,11 @@ def load_unet_sdxl_nunchaku(repo_id):
     return unet
 
 
-def load_unet(model, repo_id:str=None):
-    global loaded_unet # pylint: disable=global-statement
+def load_unet(model, repo_id: str | None = None):
+    global loaded_unet  # pylint: disable=global-statement
 
-    if ("StableDiffusionXLPipeline" in model.__class__.__name__) and (('stable-diffusion-xl-base' in repo_id) or ('sdxl-turbo' in repo_id)):
-        if model_quant.check_nunchaku('Model'):
+    if ("StableDiffusionXLPipeline" in model.__class__.__name__) and repo_id is not None and (("stable-diffusion-xl-base" in repo_id) or ("sdxl-turbo" in repo_id)):
+        if model_quant.check_nunchaku("Model"):
             unet = load_unet_sdxl_nunchaku(repo_id)
             if unet is not None:
                 model.unet = unet