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 |
|