wealthcoders commited on
Commit
bbcbe7f
·
verified ·
1 Parent(s): aafbcc8

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -24,6 +24,6 @@ class EndpointHandler:
24
  )
25
  inputs = inputs.to(self.model.device)
26
 
27
- generated_ids = self.model.generate(**inputs, max_new_tokens=128)
28
  output_text = self.processor.batch_decode(generated_ids, skip_special_tokens=True)
29
  return output_text[0]
 
24
  )
25
  inputs = inputs.to(self.model.device)
26
 
27
+ generated_ids = self.model.generate(**inputs, max_new_tokens=2048)
28
  output_text = self.processor.batch_decode(generated_ids, skip_special_tokens=True)
29
  return output_text[0]