sooh-j commited on
Commit
4ef9dc8
·
verified ·
1 Parent(s): 091eb18

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -50,7 +50,7 @@ class EndpointHandler():
50
 
51
  with torch.no_grad():
52
  out = self.model.generate(**processed,
53
- max_new_tokens=50,
54
  temperature = 0.5,
55
  do_sample=True,
56
  top_k=50,
 
50
 
51
  with torch.no_grad():
52
  out = self.model.generate(**processed,
53
+ max_new_tokens=20,
54
  temperature = 0.5,
55
  do_sample=True,
56
  top_k=50,