NeuraCraft commited on
Commit
963ddf8
·
verified ·
1 Parent(s): 098e4d7

Update lance_ai_model.py

Browse files
Files changed (1) hide show
  1. lance_ai_model.py +2 -0
lance_ai_model.py CHANGED
@@ -24,7 +24,9 @@ class LanceAIConfig(PretrainedConfig):
24
  super().__init__(**kwargs)
25
  self.vocab_size = vocab_size
26
  self.hidden_size = hidden_size
 
27
  self.num_layers = num_layers
 
28
  self.num_heads = num_heads
29
  self.intermediate_size = intermediate_size
30
  self.hidden_dropout = hidden_dropout
 
24
  super().__init__(**kwargs)
25
  self.vocab_size = vocab_size
26
  self.hidden_size = hidden_size
27
+ self.num_hidden_layers = num_layers
28
  self.num_layers = num_layers
29
+ self.num_attention_heads = num_heads
30
  self.num_heads = num_heads
31
  self.intermediate_size = intermediate_size
32
  self.hidden_dropout = hidden_dropout