Update configuration_steerling.py
Browse files
configuration_steerling.py
CHANGED
|
@@ -7,6 +7,7 @@ class SteerlingConfig(PretrainedConfig):
|
|
| 7 |
def __init__(
|
| 8 |
self,
|
| 9 |
vocab_size=100352,
|
|
|
|
| 10 |
interpretable=True,
|
| 11 |
n_layers=32,
|
| 12 |
n_head=32,
|
|
@@ -54,6 +55,7 @@ class SteerlingConfig(PretrainedConfig):
|
|
| 54 |
eot_id=100283,
|
| 55 |
**kwargs,
|
| 56 |
):
|
|
|
|
| 57 |
self.interpretable = interpretable
|
| 58 |
self.n_layers = n_layers
|
| 59 |
self.n_head = n_head
|
|
|
|
| 7 |
def __init__(
|
| 8 |
self,
|
| 9 |
vocab_size=100352,
|
| 10 |
+
instruct=False,
|
| 11 |
interpretable=True,
|
| 12 |
n_layers=32,
|
| 13 |
n_head=32,
|
|
|
|
| 55 |
eot_id=100283,
|
| 56 |
**kwargs,
|
| 57 |
):
|
| 58 |
+
self.instruct = instruct
|
| 59 |
self.interpretable = interpretable
|
| 60 |
self.n_layers = n_layers
|
| 61 |
self.n_head = n_head
|