Skip to content

GitLab

"...git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "6c7ff9c7937309b92fe8ff8a74b5dd0513bc7428"
Commit a03dde19 authored by comfyanonymous's avatar comfyanonymous
Browse files

Cap maximum history size at 10000. Delete oldest entry when reached.

parent 31c5ea7b
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
execution.py
View file @ a03dde19
...@@ -681,6 +681,7 @@ def validate_prompt(prompt): ...@@ -681,6 +681,7 @@ def validate_prompt(prompt):
return (True, None, list(good_outputs), node_errors) return (True, None, list(good_outputs), node_errors)
MAXIMUM_HISTORY_SIZE = 10000
class PromptQueue: class PromptQueue:
def __init__(self, server): def __init__(self, server):
...@@ -713,6 +714,8 @@ class PromptQueue: ...@@ -713,6 +714,8 @@ class PromptQueue:
def task_done(self, item_id, outputs): def task_done(self, item_id, outputs):
with self.mutex: with self.mutex:
prompt = self.currently_running.pop(item_id) prompt = self.currently_running.pop(item_id)
if len(self.history) > MAXIMUM_HISTORY_SIZE:
self.history.pop(next(iter(self.history)))
self.history[prompt[1]] = { "prompt": prompt, "outputs": {} } self.history[prompt[1]] = { "prompt": prompt, "outputs": {} }
for o in outputs: for o in outputs:
self.history[prompt[1]]["outputs"][o] = outputs[o] self.history[prompt[1]]["outputs"][o] = outputs[o]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment