julien-c
/

fofr-sdxl-emoji

stable-diffusion

Inference Endpoints

Model card Files Files and versions Community

multimodalart HF staff commited on Jun 25

Commit

d958ce8

•

1 Parent(s): a5e7022

Update handler.py

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

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

         inputs = data.pop("inputs", data)
         # Automatically add trigger tokens to the beginning of the prompt
+        print(data)
         images = self.pipe(
             inputs,
+            **data['parameters']
         ).images
         image = images[0]