Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -31,7 +31,7 @@ cls_model_name = "saiteki-kai/QA-DeBERTa-v3-large-binary-3"
|
|
| 31 |
|
| 32 |
model = transformers.pipeline(
|
| 33 |
model=chat_model_name,
|
| 34 |
-
model_kwargs={"dtype": torch.bfloat16}
|
| 35 |
device=device,
|
| 36 |
)
|
| 37 |
model.tokenizer.padding_side = "left"
|
|
|
|
| 31 |
|
| 32 |
model = transformers.pipeline(
|
| 33 |
model=chat_model_name,
|
| 34 |
+
model_kwargs={"dtype": torch.bfloat16},
|
| 35 |
device=device,
|
| 36 |
)
|
| 37 |
model.tokenizer.padding_side = "left"
|