wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-06 11:32:31 +08:00
Code Issues Actions 18 Packages Projects Releases Wiki Activity

ruff

Browse Source
This commit is contained in:
Rattus 2026-01-21 14:43:14 +10:00
parent aef8d003c9
commit 96e5d45642
3 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/k_diffusion/sampling.py
View File

@ -18,7 +18,7 @@ import comfy.memory_management
def trange(*args, **kwargs): def trange(*args, **kwargs):
if comfy.memory_management.aimdo_allocator == None: if comfy.memory_management.aimdo_allocator is None:
return trange_(*args, **kwargs) return trange_(*args, **kwargs)
pbar = trange_(*args, **kwargs, smoothing=1.0) pbar = trange_(*args, **kwargs, smoothing=1.0)
@ -26,7 +26,6 @@ def trange(*args, **kwargs):
pbar.set_postfix_str(" Model Initializing ... ") pbar.set_postfix_str(" Model Initializing ... ")
_update = pbar.update _update = pbar.update
initialized = False
def warmup_update(n=1): def warmup_update(n=1):
pbar._i += 1 pbar._i += 1

4
comfy/utils.py
View File

@ -32,7 +32,6 @@ from comfy.cli_args import args, enables_dynamic_vram
import json import json
import time import time
import mmap import mmap
import ctypes
import packaging import packaging
@ -96,7 +95,8 @@ def load_safetensors(ckpt):
sd = {} sd = {}
for name, info in header.items(): for name, info in header.items():
if name == "__metadata__": continue if name == "__metadata__":
continue
start, end = info["data_offsets"] start, end = info["data_offsets"]
sd[name] = data_area[start:end].view(_TYPES[info["dtype"]]).view(info["shape"]) sd[name] = data_area[start:end].view(_TYPES[info["dtype"]]).view(info["shape"])

1
execution.py
View File

@ -1,4 +1,3 @@
import gc
import copy import copy
import heapq import heapq
import inspect import inspect