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