Upload modeling_aquila.py
Browse files- modeling_aquila.py +1 -1
modeling_aquila.py
CHANGED
@@ -695,7 +695,7 @@ class LlamaModel(LlamaPreTrainedModel):
|
|
695 |
)
|
696 |
|
697 |
|
698 |
-
class
|
699 |
def __init__(self, config):
|
700 |
super().__init__(config)
|
701 |
self.model = LlamaModel(config)
|
|
|
695 |
)
|
696 |
|
697 |
|
698 |
+
class AquilaForCausalLM(LlamaPreTrainedModel):
|
699 |
def __init__(self, config):
|
700 |
super().__init__(config)
|
701 |
self.model = LlamaModel(config)
|