multimodalart HF staff commited on
Commit
b3d9998
1 Parent(s): f1f2c1f

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +0 -2
handler.py CHANGED
@@ -44,10 +44,8 @@ class EndpointHandler:
44
  inputs = data.pop("inputs", data)
45
 
46
  # Automatically add trigger tokens to the beginning of the prompt
47
- print(data)
48
  images = self.pipe(
49
  inputs,
50
- cross_attention_kwargs={"scale": 0.8},
51
  **data['parameters']
52
  ).images
53
  image = images[0]
 
44
  inputs = data.pop("inputs", data)
45
 
46
  # Automatically add trigger tokens to the beginning of the prompt
 
47
  images = self.pipe(
48
  inputs,
 
49
  **data['parameters']
50
  ).images
51
  image = images[0]