Update handler.py

handler.py

@@ -44,10 +44,10 @@ class EndpointHandler:
         inputs = data.pop("inputs", data)
 
         # Automatically add trigger tokens to the beginning of the prompt
+        print(data)
         images = self.pipe(
             inputs,
-            cross_attention_kwargs={"scale": 0.8},
-            num_inference_steps=25
+            **data['parameters']
         ).images
         image = images[0]