empty cache after execute()

3a83da72 · masterpiecebestquality · 00a9189e · 3a83da72
Commit 3a83da72 authored Feb 20, 2023 by masterpiecebestquality
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

main.py main.py +1 -0

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