lea97338 commited on
Commit
685afd8
·
verified ·
1 Parent(s): d06be0d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -18,9 +18,7 @@ def encode_text(prompt: str):
18
  inputs = tokenizer(
19
  prompt,
20
  return_tensors="pt",
21
- padding=True,
22
- truncation=True,
23
- max_length=1024,
24
  )
25
 
26
  with torch.inference_mode():
 
18
  inputs = tokenizer(
19
  prompt,
20
  return_tensors="pt",
21
+ max_length=512,
 
 
22
  )
23
 
24
  with torch.inference_mode():