Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3# 모델과 토크나이저 로드
4tokenizer = AutoTokenizer.from_pretrained("himedia/fincredit-lamma-3.2-3b-lr2e04-bs16-r64-steps10-20250702_181705")
5model = AutoModelForCausalLM.from_pretrained("himedia/fincredit-lamma-3.2-3b-lr2e04-bs16-r64-steps10-20250702_181705")
6
7# 간단한 추론 예제
8prompt = "고객의 신용등급을 평가해주세요:"
9inputs = tokenizer(prompt, return_tensors="pt")
10outputs = model.generate(**inputs, max_length=200)
11result = tokenizer.decode(outputs[0], skip_special_tokens=True)
12print(result)1from vllm import LLM, SamplingParams
2
3# vLLM 로드 (병합된 모델이므로 바로 사용 가능)
4llm = LLM(
5 model="himedia/fincredit-lamma-3.2-3b-lr2e04-bs16-r64-steps10-20250702_181705",
6 max_model_len=2048,
7 gpu_memory_utilization=0.85
8)
9
10# 샘플링 파라미터 설정
11sampling_params = SamplingParams(
12 temperature=0.7,
13 top_p=0.9,
14 max_tokens=200
15)
16
17# 추론
18prompts = ["고객의 신용등급을 평가해주세요:"]
19outputs = llm.generate(prompts, sampling_params)
20
21for output in outputs:
22 prompt = output.prompt
23 generated_text = output.outputs[0].text
24 print(f"Prompt: {prompt!r}")
25 print(f"Generated text: {generated_text!r}")1from unsloth import FastLanguageModel
2
3# 원본 LoRA 어댑터로 테스트
4model, tokenizer = FastLanguageModel.from_pretrained(
5 model_name = "himedia/fincredit-Llama-3.2-3B-lr2e04-bs16-r64-steps1000-20250623_060351", # LoRA 어댑터
6 max_seq_length = 2048,
7 dtype = None,
8 load_in_4bit = True,
9)training_log.json: 전체 학습 로그 (JSON 형식)FinCreditLlama-3.2-3B_20250702_181705_training_curves.png: 학습 곡선 시각화 이미지fincredit-lamma-3.2-3b-lr2e04-bs16-r64-steps10-20250702_181705 = fincredit-lamma3-4b-lr2e04-bs2-r64-steps10-20250702_181705fincredit-lamma3-4b: 모델 기본명lr2e04: Learning Ratebs2: Batch Sizer64: LoRA ranksteps10: 학습 스텝20250702_181705: 학습 시각himedia/fincredit-Llama-3.2-3B-lr2e04-bs16-r64-steps1000-20250623_060351