from transformers import AutoModelForCausalLM, AutoTokenizer
import torch

path = "mssma/ko-solar-10.7b-v0.4"
model = AutoModelForCausalLM.from_pretrained(
        path,
        return_dict=True,
        torch_dtype=torch.float16,
        device_map='auto'
)
tokenizer = AutoTokenizer.from_pretrained(path)