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(
|