298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-20 03:13:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

empty cache after execute()

Browse Source
This commit is contained in:
masterpiecebestquality 2023-02-20 13:56:34 -05:00
parent 00a9189e30
commit 3a83da7281
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
main.py
View File

@ -209,6 +209,7 @@ class PromptExecutor:
executed = set(executed) executed = set(executed)
for x in executed: for x in executed:
self.old_prompt[x] = copy.deepcopy(prompt[x]) self.old_prompt[x] = copy.deepcopy(prompt[x])
torch.cuda.empty_cache()
def validate_inputs(prompt, item): def validate_inputs(prompt, item):
unique_id = item unique_id = item