lea97338 commited on
Commit
d06be0d
·
verified ·
1 Parent(s): 4aac80a

Update app.py

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