Gagan Bhatia commited on
Commit
a2f8e94
1 Parent(s): 027f3d0

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +4 -0
src/models/model.py CHANGED
@@ -65,3 +65,7 @@ class DataModule(Dataset):
65
  padding="max_length",
66
  truncation=True,
67
  return_attention_mask=True,
 
 
 
 
 
65
  padding="max_length",
66
  truncation=True,
67
  return_attention_mask=True,
68
+ add_special_tokens=True,
69
+ return_tensors="pt",
70
+ )
71
+