restore assert info
Browse files- modeling_internlm.py +3 -0
modeling_internlm.py
CHANGED
|
@@ -532,6 +532,9 @@ class InternLMModel(InternLMPreTrainedModel):
|
|
| 532 |
_auto_class = "AutoModel"
|
| 533 |
|
| 534 |
def __init__(self, config: InternLMConfig):
|
|
|
|
|
|
|
|
|
|
| 535 |
super().__init__(config)
|
| 536 |
self.padding_idx = config.pad_token_id
|
| 537 |
self.vocab_size = config.vocab_size
|
|
|
|
| 532 |
_auto_class = "AutoModel"
|
| 533 |
|
| 534 |
def __init__(self, config: InternLMConfig):
|
| 535 |
+
assert (0), 'Inference by transformers is currently not supported, ' \
|
| 536 |
+
'please follow README to convert the model ' \
|
| 537 |
+
'and use lmdeploy (https://github.com/InternLM/lmdeploy) for inference.'
|
| 538 |
super().__init__(config)
|
| 539 |
self.padding_idx = config.pad_token_id
|
| 540 |
self.vocab_size = config.vocab_size
|