Update handler.py

handler.py

@@ -22,10 +22,7 @@ class EndpointHandler():
                                                               safety_checker=None)
         self.pipe.load_lora_weights("Oysiyl/sdxl-lora-android-google-toy", weights="pytorch_lora_weights.safetensors")
         self.pipe.enable_xformers_memory_efficient_attention()
-        self.pipe = self.pipe.to(device)
-        self.seed = 42
-        # Define Generator with seed
-        self.generator = torch.Generator(device="cpu").manual_seed(self.seed)
+        self.pipe.to(device)
 
     def __call__(self, data: Any) -> List[List[Dict[str, float]]]:
         """
@@ -38,13 +35,8 @@ class EndpointHandler():
         # Check if prompt is not provided
         if prompt is None:
             return {"error": "Please provide a prompt."}
-        
-        # Check if seed changed
-        if seed is not None and seed != self.seed:
-            print(f"changing seed from {self.seed} to {seed}")
-            self.seed = seed
-            self.generator = torch.Generator(device="cpu").manual_seed(self.seed)
-        
+
+        generator = torch.Generator(device="cpu").manual_seed(self.seed)      
         
         # hyperparamters
         num_inference_steps = data.pop("num_inference_steps", 50)
@@ -59,7 +51,8 @@ class EndpointHandler():
             guidance_scale=guidance_scale,
             temperature=temperature,
             num_images_per_prompt=1,
-            generator=self.generator
+            seed=seed,
+            generator=generator
         )