File size: 210 Bytes
ab192f6
 
 
 
 
 
1
2
3
4
5
6
from transformers.models.llama.configuration_llama import LlamaConfig

class MyLlamaConfig(LlamaConfig):
    model_type = "custom_llama"
    def __init__(self, **kwargs):
        super().__init__(**kwargs)