MindLabUnimib commited on
Commit
ead4659
·
verified ·
1 Parent(s): 81477d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -58,7 +58,7 @@ cls_model_name = "saiteki-kai/QA-DeBERTa-v3-large-binary-3"
58
 
59
  model = transformers.pipeline(
60
  model=chat_model_name,
61
- model_kwargs={"dtype": torch.bfloat16} | ({"attn_implementation": "flash_attention_2"} if is_flash_attn_2_available() else {}),
62
  device=device,
63
  )
64
 
 
58
 
59
  model = transformers.pipeline(
60
  model=chat_model_name,
61
+ model_kwargs={"dtype": torch.bfloat16, "padding_side": "left"} | ({"attn_implementation": "flash_attention_2"} if is_flash_attn_2_available() else {}),
62
  device=device,
63
  )
64