EcomGen-0.0.2v

상품명 기반 상품설명 생성 모델

모델 개요

이 모델은 kakaocorp/kanana-nano-2.1b-instruct를 기반으로 fine-tuning된 상품설명 생성 모델입니다. 상품명을 입력으로 받아 해당 상품에 대한 상세한 설명을 자동으로 생성합니다.

주요 특징

기반 모델: kakaocorp/kanana-nano-2.1b-instruct

fine-tuning 데이터: 다양한 카테고리의 상품명과 상품설명 페어 데이터

ex)
입력: 상품명
출력: 생성된 상품설명

• 언어: 한국어

• max_token_length : 512

• 사용방법

# Load model directly
model_path = "UICHEOL-HWANG/EcomGen-0.0.2v"


model = AutoModelForCausalLM.from_pretrained(
    model_path,
    torch_dtype=torch.bfloat16,
    trust_remote_code=True,
).to("cuda")

tokenizer = AutoTokenizer.from_pretrained(model_path)

prompt = "주어진 상품명을 기반으로 매력적인 상품설명을 작성하세요.\n상품명: 김갑생할머니김"
messages = [
    {"role": "system", "content": "당신은 상품명을 바탕으로 상세하고 매력적인 상품 설명을 생성하는 AI입니다."},
    {"role": "user", "content": prompt}
]

input_ids = tokenizer.apply_chat_template(
    messages,
    tokenize=True,
    add_generation_prompt=True,
    return_tensors="pt"
).to("cuda")

_ = model.eval()
with torch.no_grad():
    output = model.generate(
        input_ids,
        max_new_tokens=512,
        do_sample=False,
    )

print(tokenizer.decode(output[0], skip_special_tokens=True))