From 9ab14711e5fcbe79fb040820cc080b58d3d06a15 Mon Sep 17 00:00:00 2001
From: catboxanon <122327233+catboxanon@users.noreply.github.com>
Date: Mon, 10 Feb 2025 15:12:54 -0500
Subject: [PATCH] Remove deepcopy usage in PromptQueue

---
 execution.py | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

diff --git a/execution.py b/execution.py
index 2c979205b..e84ae59b5 100644
--- a/execution.py
+++ b/execution.py
@@ -1,5 +1,4 @@
 import sys
-import copy
 import logging
 import threading
 import heapq
@@ -880,6 +879,16 @@ class PromptQueue:
         self.flags = {}
         server.prompt_queue = self
 
+    def _copy(self, obj):
+        if isinstance(obj, dict):
+            return {k: self._copy(v) for k, v in obj.items()}
+        elif isinstance(obj, tuple):
+            return tuple(self._copy(v) for v in obj)
+        elif isinstance(obj, list):
+            return [self._copy(v) for v in obj]
+        else:
+            return obj
+
     def put(self, item):
         with self.mutex:
             heapq.heappush(self.queue, item)
@@ -894,7 +903,7 @@ class PromptQueue:
                     return None
             item = heapq.heappop(self.queue)
             i = self.task_counter
-            self.currently_running[i] = copy.deepcopy(item)
+            self.currently_running[i] = self._copy(item)
             self.task_counter += 1
             self.server.queue_updated()
             return (item, i)
@@ -913,7 +922,7 @@ class PromptQueue:
 
             status_dict: Optional[dict] = None
             if status is not None:
-                status_dict = copy.deepcopy(status._asdict())
+                status_dict = self._copy(status._asdict())
 
             self.history[prompt[1]] = {
                 "prompt": prompt,
@@ -928,7 +937,7 @@ class PromptQueue:
             out = []
             for x in self.currently_running.values():
                 out += [x]
-            return (out, copy.deepcopy(self.queue))
+            return (out, self._copy(self.queue))
 
     def get_tasks_remaining(self):
         with self.mutex:
@@ -967,7 +976,7 @@ class PromptQueue:
                     i += 1
                 return out
             elif prompt_id in self.history:
-                return {prompt_id: copy.deepcopy(self.history[prompt_id])}
+                return {prompt_id: self._copy(self.history[prompt_id])}
             else:
                 return {}