forced lora

handler.py

@@ -88,7 +88,7 @@ class EndpointHandler:
         self.inference_progress = {}  # Dictionary to store progress of each request
         self.inference_images = {}  # Dictionary to store latest image of each request
         self.total_steps = {}
-        self.active_request_ids = set()
+        self.active_request_ids = list()
         self.inference_in_progress = False
 
         self.executor = ThreadPoolExecutor(
@@ -260,20 +260,20 @@ class EndpointHandler:
 
         return negative_prompt
 
-    def clean_request_data(self, request_id: str):
+    def clean_request_data(self):
         """Clean up the data related to a specific request ID."""
 
         # Remove the request ID from the progress dictionary
-        self.inference_progress.pop(request_id, None)
+        self.inference_progress.clear()
 
         # Remove the request ID from the images dictionary
-        self.inference_images.pop(request_id, None)
+        self.inference_images.clear()
 
         # Remove the request ID from the total_steps dictionary
-        self.total_steps.pop(request_id, None)
+        self.total_steps.clear()
 
         # Delete request id
-        self.active_request_ids.discard(request_id)
+        self.active_request_ids.clear()
 
         # Set inference to False
         self.inference_in_progress = False
@@ -294,7 +294,11 @@ class EndpointHandler:
                 img = Image.fromarray(img_data, "RGB")
 
                 # Apply a blur to the image
-                img = img.filter(ImageFilter.GaussianBlur(radius=5))
+                # more intense at the beginning
+                if step < int(self.total_steps[self.active_request_ids[0]] / 1.5):
+                    img = img.filter(ImageFilter.GaussianBlur(radius=30))
+                else:
+                    img = img.filter(ImageFilter.GaussianBlur(radius=10))
 
                 # print(img_data)
             else:
@@ -462,11 +466,11 @@ class EndpointHandler:
                 }
 
             # Set the inference state to in progress
-            self.clean_request_data(request_id)
+            self.clean_request_data()
             self.inference_in_progress = True
             self.inference_progress[request_id] = 0
             self.inference_images[request_id] = None
-            self.active_request_ids.add(request_id)
+            self.active_request_ids.append(request_id)
 
             self.executor.submit(self.start_inference, data)