Update reader.py

aditya0by0 · aditya0by0 · commit ceb4e70cb9c7 · 2024-10-31T14:02:38.000+01:00
diff --git a/chebai/preprocessing/reader.py b/chebai/preprocessing/reader.py
@@ -8,9 +8,9 @@
 
 from chebai.preprocessing.collate import DefaultCollator, RaggedCollator
 
-EMBEDDING_OFFSET = 100
+EMBEDDING_OFFSET = 10
 PADDING_TOKEN_INDEX = 0
-MASK_TOKEN_INDEX = 1
+MASK_TOKEN_INDEX = 10
 CLS_TOKEN = 2
 
 
