bambadij commited on
Commit
51ae396
·
verified ·
1 Parent(s): 79b577f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -64,7 +64,7 @@ async def predict(request: PredictionRequest):
64
  prompt = default_prompt
65
 
66
  # Tokenize l'entrée et créez un attention mask
67
- inputs = tokenizer(prompt, return_tensors="pt",tokenizer.pad_token == tokenizer.eos_token, padding=True, truncation=True)
68
  input_ids = inputs.input_ids.to(model.device)
69
  attention_mask = inputs.attention_mask.to(model.device)
70
 
 
64
  prompt = default_prompt
65
 
66
  # Tokenize l'entrée et créez un attention mask
67
+ inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
68
  input_ids = inputs.input_ids.to(model.device)
69
  attention_mask = inputs.attention_mask.to(model.device)
70