qnloft
/
ComfyUI
mirror of https://github.com/comfyanonymous/ComfyUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Further tuning and fix mem_free_total.

This commit is contained in:
Simon Lui 2023-08-19 21:35:22 -07:00 committed by comfyanonymous
parent 2c096e4260
commit 9225465975
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/model_management.py
View File

@ -271,8 +271,7 @@ class LoadedModel:
self.model_accelerated = True
if xpu_available and not args.disable_ipex_optimize:
self.real_model.training = False
self.real_model = torch.xpu.optimize(self.real_model, inplace=True)
self.real_model = torch.xpu.optimize(self.real_model.eval(), inplace=True, auto_kernel_selection=True, graph_mode=True)
return self.real_model
@ -515,7 +514,7 @@ def get_free_memory(dev=None, torch_free_too=False):
mem_allocated = stats['allocated_bytes.all.current']
mem_reserved = stats['reserved_bytes.all.current']
mem_free_torch = mem_reserved - mem_active
mem_free_total = torch.xpu.get_device_properties(dev).total_memory - mem_allocated + mem_free_torch
mem_free_total = torch.xpu.get_device_properties(dev).total_memory - mem_allocated
else:
stats = torch.cuda.memory_stats(dev)
mem_active = stats['active_bytes.all.current']