wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-23 12:50:18 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Fix linting issues

Browse Source
This commit is contained in:
doctorpangloss 2025-01-04 14:07:19 -08:00
parent f8a00af0d2
commit 12082d877d
1 changed files with 4 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
comfy/model_management.py
View File

@ -25,7 +25,7 @@ import warnings
import weakref import weakref
from enum import Enum from enum import Enum
from threading import RLock from threading import RLock
from typing import List, Sequence, Final from typing import List, Sequence, Final, Optional
import psutil import psutil
import torch import torch
@ -110,7 +110,7 @@ except:
pass pass
try: try:
import torch_npu # noqa: F401 import torch_npu # pylint: disable=import-error, noqa: F401
_ = torch.npu.device_count() _ = torch.npu.device_count()
npu_available = torch.npu.is_available() npu_available = torch.npu.is_available()
@ -257,14 +257,6 @@ def is_amd():
return False return False
def is_amd():
global cpu_state
if cpu_state == CPUState.GPU:
if torch.version.hip:
return True
return False
MIN_WEIGHT_MEMORY_RATIO = 0.4 MIN_WEIGHT_MEMORY_RATIO = 0.4
if is_nvidia(): if is_nvidia():
MIN_WEIGHT_MEMORY_RATIO = 0.2 MIN_WEIGHT_MEMORY_RATIO = 0.2
@ -292,7 +284,7 @@ if ENABLE_PYTORCH_ATTENTION:
try: try:
if int(torch_version[0]) == 2 and int(torch_version[2]) >= 5: if int(torch_version[0]) == 2 and int(torch_version[2]) >= 5:
torch.backends.cuda.allow_fp16_bf16_reduction_math_sdp(True) torch.backends.cuda.allow_fp16_bf16_reduction_math_sdp(True) # pylint: disable=no-member
except: except:
logging.warning("Warning, could not set allow_fp16_bf16_reduction_math_sdp") logging.warning("Warning, could not set allow_fp16_bf16_reduction_math_sdp")
@ -1092,7 +1084,7 @@ def get_free_memory(dev=None, torch_free_too=False):
mem_free_total = 1024 * 1024 * 1024 # TODO mem_free_total = 1024 * 1024 * 1024 # TODO
mem_free_torch = mem_free_total mem_free_torch = mem_free_total
elif is_intel_xpu(): elif is_intel_xpu():
stats = torch.xpu.memory_stats(dev) stats = torch.xpu.memory_stats(dev) # pylint: disable=no-member
mem_active = stats['active_bytes.all.current'] mem_active = stats['active_bytes.all.current']
mem_reserved = stats['reserved_bytes.all.current'] mem_reserved = stats['reserved_bytes.all.current']
mem_free_torch = mem_reserved - mem_active mem_free_torch = mem_reserved - mem_active