Fix class definition
Browse files- modeling_llama.py +1 -1
modeling_llama.py
CHANGED
|
@@ -58,7 +58,7 @@ logger = logging.get_logger(__name__)
|
|
| 58 |
_CONFIG_FOR_DOC = "LlamaConfig"
|
| 59 |
|
| 60 |
|
| 61 |
-
CustomLlamaConfig(LlamaConfig):
|
| 62 |
model_type = "custom_llama"
|
| 63 |
|
| 64 |
|
|
|
|
| 58 |
_CONFIG_FOR_DOC = "LlamaConfig"
|
| 59 |
|
| 60 |
|
| 61 |
+
class CustomLlamaConfig(LlamaConfig):
|
| 62 |
model_type = "custom_llama"
|
| 63 |
|
| 64 |
|