oberbics commited on
Commit
a87e4b6
·
verified ·
1 Parent(s): 62cad6a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -88,7 +88,7 @@ Extract explicit and implicit arguments from historical text.
88
  {text}<|eot_id|>
89
  <|start_header_id|>assistant<|end_header_id|>"""
90
 
91
- inputs = tokenizer(prompt, return_tensors="pt", max_length=5048).to(model.device)
92
  input_length = inputs['input_ids'].shape[1]
93
  try:
94
  if temperature is None:
 
88
  {text}<|eot_id|>
89
  <|start_header_id|>assistant<|end_header_id|>"""
90
 
91
+ inputs = tokenizer(prompt, return_tensors="pt",truncation=True, max_length=5048).to(model.device)
92
  input_length = inputs['input_ids'].shape[1]
93
  try:
94
  if temperature is None: