[HuggingFace] Peft 모델 torch 형식으로 업로드하기 (push_to_hub)

jongmin-oh·2023년 11월 30일
0

Prepare

import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
from peft import PeftModel

from huggingface_hub import login
login(token="{your_write_token}")

Merge Model

model_id = "yanolja/EEVE-Korean-10.8B-v1.0"
base_model = AutoModelForCausalLM.from_pretrained(
    model_id,
    return_dict=True,
    torch_dtype=torch.float16
)
tokenizer = AutoTokenizer.from_pretrained(model_id)
upload_model = PeftModel.from_pretrained(base_model, "{peft save path}")
upload_model = upload_model.merge_and_unload()

Upload

upload_model.push_to_hub("j5ng/{model-name}", use_temp_dir=True)
tokenizer.push_to_hub("j5ng/{model-name}", use_temp_dir=True)
profile
스타트업에서 자연어처리 챗봇을 연구하는 머신러닝 개발자입니다.

0개의 댓글