Merge branch 'ipex' of https://github.com/kwaa/ComfyUI-IPEX
This commit is contained in:
commit
f4c689ea89
|
@ -10,6 +10,8 @@ import gc
|
|||
import torch
|
||||
import nodes
|
||||
|
||||
from model_management import xpu_available
|
||||
|
||||
def get_input_data(inputs, class_def, unique_id, outputs={}, prompt={}, extra_data={}):
|
||||
valid_inputs = class_def.INPUT_TYPES()
|
||||
input_data_all = {}
|
||||
|
@ -206,6 +208,8 @@ class PromptExecutor:
|
|||
if torch.version.cuda: #This seems to make things worse on ROCm so I only do it for cuda
|
||||
torch.cuda.empty_cache()
|
||||
torch.cuda.ipc_collect()
|
||||
elif xpu_available:
|
||||
torch.xpu.empty_cache()
|
||||
|
||||
|
||||
def validate_inputs(prompt, item):
|
||||
|
|
Loading…
Reference in New Issue