From 96e5d456422bd3c50b7f735c61ec40aa8ec7c6d8 Mon Sep 17 00:00:00 2001 From: Rattus Date: Wed, 21 Jan 2026 14:43:14 +1000 Subject: [PATCH] ruff --- comfy/k_diffusion/sampling.py | 3 +-- comfy/utils.py | 4 ++-- execution.py | 1 - 3 files changed, 3 insertions(+), 5 deletions(-) diff --git a/comfy/k_diffusion/sampling.py b/comfy/k_diffusion/sampling.py index 2a08066a0..c0c51d51a 100644 --- a/comfy/k_diffusion/sampling.py +++ b/comfy/k_diffusion/sampling.py @@ -18,7 +18,7 @@ import comfy.memory_management def trange(*args, **kwargs): - if comfy.memory_management.aimdo_allocator == None: + if comfy.memory_management.aimdo_allocator is None: return trange_(*args, **kwargs) pbar = trange_(*args, **kwargs, smoothing=1.0) @@ -26,7 +26,6 @@ def trange(*args, **kwargs): pbar.set_postfix_str(" Model Initializing ... ") _update = pbar.update - initialized = False def warmup_update(n=1): pbar._i += 1 diff --git a/comfy/utils.py b/comfy/utils.py index 9471eed20..c620e7545 100644 --- a/comfy/utils.py +++ b/comfy/utils.py @@ -32,7 +32,6 @@ from comfy.cli_args import args, enables_dynamic_vram import json import time import mmap -import ctypes import packaging @@ -96,7 +95,8 @@ def load_safetensors(ckpt): sd = {} for name, info in header.items(): - if name == "__metadata__": continue + if name == "__metadata__": + continue start, end = info["data_offsets"] sd[name] = data_area[start:end].view(_TYPES[info["dtype"]]).view(info["shape"]) diff --git a/execution.py b/execution.py index 4148baaf5..a25bd36cd 100644 --- a/execution.py +++ b/execution.py @@ -1,4 +1,3 @@ -import gc import copy import heapq import inspect