|
|
|
@ -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): |
|
|
|
|