sagu7 commited on
Commit
da0e8a7
·
1 Parent(s): 849a8db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -91,7 +91,7 @@ async def evaluate(
91
  max_new_tokens=128,
92
  seed=None,
93
  do_sample=True,
94
- **kwargs,
95
  ):
96
  prompt = generate_prompt(input)
97
  inputs = tokenizer(prompt, return_tensors="pt")
@@ -101,7 +101,7 @@ async def evaluate(
101
  top_p=top_p,
102
  top_k=top_k,
103
  num_beams=num_beams,
104
- **kwargs,
105
  )
106
  with torch.no_grad():
107
  generation_output = model.generate(
 
91
  max_new_tokens=128,
92
  seed=None,
93
  do_sample=True,
94
+ # **kwargs,
95
  ):
96
  prompt = generate_prompt(input)
97
  inputs = tokenizer(prompt, return_tensors="pt")
 
101
  top_p=top_p,
102
  top_k=top_k,
103
  num_beams=num_beams,
104
+ # **kwargs,
105
  )
106
  with torch.no_grad():
107
  generation_output = model.generate(