xiaohua828
commited on
Commit
·
5ae1029
1
Parent(s):
d80aaea
Update config_counter.py
Browse files- config_counter.py +2 -2
config_counter.py
CHANGED
@@ -17,8 +17,8 @@ class GLMConfig(PretrainedConfig):
|
|
17 |
**kwargs
|
18 |
):
|
19 |
|
20 |
-
self.
|
21 |
-
self.
|
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)
|