sooh-j commited on
Commit
27e36d4
1 Parent(s): 1210955

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +4 -4
handler.py CHANGED
@@ -51,10 +51,10 @@ class EndpointHandler():
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,
57
- top_p=0.9,
58
  repetition_penalty=1.2
59
  ).to(self.device)
60
 
 
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,
57
+ # top_p=0.9,
58
  repetition_penalty=1.2
59
  ).to(self.device)
60