Darayut commited on
Commit
b81a46e
·
verified ·
1 Parent(s): ba55463

Upload configuration_khmerocr.py

Browse files
Files changed (1) hide show
  1. configuration_khmerocr.py +25 -0
configuration_khmerocr.py ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # configuration_khmerocr.py
2
+ from transformers import PretrainedConfig
3
+
4
+ class KhmerOCRConfig(PretrainedConfig):
5
+ model_type = "khmer-ocr"
6
+
7
+ def __init__(
8
+ self,
9
+ vocab_size=124,
10
+ emb_dim=384,
11
+ max_global_len=4096,
12
+ pad_idx=0,
13
+ nhead=8,
14
+ num_encoder_layers=2,
15
+ num_decoder_layers=2,
16
+ **kwargs,
17
+ ):
18
+ self.vocab_size = vocab_size
19
+ self.emb_dim = emb_dim
20
+ self.max_global_len = max_global_len
21
+ self.pad_idx = pad_idx
22
+ self.nhead = nhead
23
+ self.num_encoder_layers = num_encoder_layers
24
+ self.num_decoder_layers = num_decoder_layers
25
+ super().__init__(**kwargs)