Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
6a7186e
1
Parent(s):
2ad8ac8
Update model.py
Browse files- src/models/model.py +4 -0
src/models/model.py
CHANGED
|
@@ -71,3 +71,7 @@ class DataModule(Dataset):
|
|
| 71 |
|
| 72 |
labels = output_encoding["input_ids"]
|
| 73 |
labels[
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 71 |
|
| 72 |
labels = output_encoding["input_ids"]
|
| 73 |
labels[
|
| 74 |
+
labels == 0
|
| 75 |
+
] = -100
|
| 76 |
+
|
| 77 |
+
return dict(
|