OpenLab-NLP
/

model-prototype

Model card Files Files and versions

Yuchan commited on Nov 21

Commit

340a91d

·

verified ·

1 Parent(s): ff72f32

Update Model.py

Files changed (1) hide show

Model.py +2 -2

Model.py CHANGED Viewed

@@ -68,8 +68,8 @@ unk_id = sp.piece_to_id("<unk>")
 vocab_size = sp.get_piece_size()
 print(f"✅ Vocabulary size: {vocab_size}")
-max_len = 150
-batch_size = 128
 def text_to_ids(text):
     return sp.encode(text, out_type=int)

 vocab_size = sp.get_piece_size()
 print(f"✅ Vocabulary size: {vocab_size}")
+max_len = 512
+batch_size = 256
 def text_to_ids(text):
     return sp.encode(text, out_type=int)