layers_block_name = "model.layers"
Browse files
modeling_InternLM_XComposer.py
CHANGED
|
@@ -37,6 +37,8 @@ class InternLMXComposerForCausalLM(PreTrainedModel):
|
|
| 37 |
max_new_tokens=500,
|
| 38 |
)
|
| 39 |
|
|
|
|
|
|
|
| 40 |
def __init__(self, config):
|
| 41 |
super().__init__(config)
|
| 42 |
|
|
|
|
| 37 |
max_new_tokens=500,
|
| 38 |
)
|
| 39 |
|
| 40 |
+
layers_block_name = "model.layers"
|
| 41 |
+
|
| 42 |
def __init__(self, config):
|
| 43 |
super().__init__(config)
|
| 44 |
|