From 0ad40d2b8b2db8b2f2d5b8f7fde2f35c043e3c71 Mon Sep 17 00:00:00 2001 From: vladmandic Date: Tue, 2 Dec 2025 12:25:04 +0100 Subject: [PATCH] lint Signed-off-by: vladmandic --- modules/sd_models.py | 4 ++-- modules/sdnq/loader.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/modules/sd_models.py b/modules/sd_models.py index 66048d58e..c1b4f7425 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -1,10 +1,11 @@ +from enum import Enum import sys import time import copy import inspect import logging +import os import os.path -from enum import Enum import diffusers import diffusers.loaders.single_file_utils import torch @@ -65,7 +66,6 @@ def set_huggingface_options(): else: sd_hijack_accelerate.restore_accelerate() if (shared.opts.runai_streamer_diffusers or shared.opts.runai_streamer_transformers) and (sys.platform == 'linux'): - import os log.debug(f'Loader: runai enabled chunk={os.environ["RUNAI_STREAMER_CHUNK_BYTESIZE"]} limit={os.environ["RUNAI_STREAMER_MEMORY_LIMIT"]}') sd_hijack_safetensors.hijack_safetensors(shared.opts.runai_streamer_diffusers, shared.opts.runai_streamer_transformers) else: diff --git a/modules/sdnq/loader.py b/modules/sdnq/loader.py index 7593c505c..71cd2df7a 100644 --- a/modules/sdnq/loader.py +++ b/modules/sdnq/loader.py @@ -120,7 +120,7 @@ def load_sdnq_model(model_path: str, model_cls: ModelMixin = None, file_name: st state_dict = load_files(files, key_mapping=key_mapping, device=device, method=load_method) if isinstance(getattr(model, "_tied_weights_keys", None), dict): - for key, value in model._tied_weights_keys.items(): + for key, value in model._tied_weights_keys.items(): # pylint: disable=protected-access if value in state_dict.keys() and key not in state_dict.keys(): state_dict[key] = state_dict[value] else: