Views
No views yet
Qwen/Qwen2.5-0.5B를 LoRA 파인튜닝한 모델입니다.
공정 상황(지시+환경)을 입력하면 로봇의 단계별 추론과 동작을 생성합니다.Qwen/Qwen2.5-0.5B1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3from peft import PeftModel
4
5base = "Qwen/Qwen2.5-0.5B"
6tok = AutoTokenizer.from_pretrained("YMmim/qwen2.5-0.5b-robot-cot-lora")
7model = AutoModelForCausalLM.from_pretrained(base, torch_dtype=torch.float16)
8model = PeftModel.from_pretrained(model, "YMmim/qwen2.5-0.5b-robot-cot-lora")