Update modeling_internlm.py
Browse files- modeling_internlm.py +1 -0
modeling_internlm.py
CHANGED
|
@@ -615,6 +615,7 @@ class InternLMForCausalLM(InternLMPreTrainedModel):
|
|
| 615 |
_auto_class = "AutoModelForCausalLM"
|
| 616 |
|
| 617 |
def __init__(self, config):
|
|
|
|
| 618 |
super().__init__(config)
|
| 619 |
self.model = InternLMModel(config)
|
| 620 |
|
|
|
|
| 615 |
_auto_class = "AutoModelForCausalLM"
|
| 616 |
|
| 617 |
def __init__(self, config):
|
| 618 |
+
assert(0), 'Inference by transformers is currently not supported, please refer to https://huggingface.co/internlm/internlm-chat-20b-4bit#inference'
|
| 619 |
super().__init__(config)
|
| 620 |
self.model = InternLMModel(config)
|
| 621 |
|