xiaohua828 commited on
Commit
5ae1029
·
1 Parent(s): d80aaea

Update config_counter.py

Browse files
Files changed (1) hide show
  1. config_counter.py +2 -2
config_counter.py CHANGED
@@ -17,8 +17,8 @@ class GLMConfig(PretrainedConfig):
17
  **kwargs
18
  ):
19
 
20
- self.num_layers = weight
21
- self.vocab_size = bias
22
  self.torch_dtype = torch_dtype
23
  self.model_type = "counter"
24
  super().__init__(**kwargs)
 
17
  **kwargs
18
  ):
19
 
20
+ self.weight = weight
21
+ self.bias = bias
22
  self.torch_dtype = torch_dtype
23
  self.model_type = "counter"
24
  super().__init__(**kwargs)