xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-02 07:07:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix xpu function not implemented p2.

Browse Source
This commit is contained in:
comfyanonymous 2025-07-23 18:16:31 -04:00 committed by GitHub
parent 39dda1d40d
commit 20700fde8e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/model_management.py
View File

@ -1107,10 +1107,7 @@ def get_free_memory(dev=None, torch_free_too=False):
stats = torch.xpu.memory_stats(dev)
mem_active = stats['active_bytes.all.current']
mem_reserved = stats['reserved_bytes.all.current']
if torch_version_numeric < (2, 6):
mem_free_xpu = torch.xpu.get_device_properties(dev).total_memory - mem_reserved
else:
mem_free_xpu, _ = torch.xpu.mem_get_info(dev)
mem_free_xpu = torch.xpu.get_device_properties(dev).total_memory - mem_reserved
mem_free_torch = mem_reserved - mem_active
mem_free_total = mem_free_xpu + mem_free_torch
elif is_ascend_npu():