lmops

1
from logger_config import logger
2
from config import Arguments
3
from llms.gpt2 import GPT2
4

5

6
class Llama(GPT2):
7

8
    def __init__(self, args: Arguments, model_name_or_path: str = 'huggyllama/llama-7b', **kwargs):
9
        super().__init__(args, model_name_or_path, **kwargs)
10