Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|