Views
No views yet
1
2```from peft import PeftModel
3
4# 베이스 모델 로드 (분류용)
5print("베이스 모델 로딩")
6base_model_reload = AutoModelForSequenceClassification.from_pretrained(
7 "klue/bert-base",
8 num_labels=2
9)
10
11# 업로드한 LoRA 어댑터 로드
12print(f"LoRA 어댑터 로딩: ellie9889/nsmc-sentiment-lora")
13model_reload = PeftModel.from_pretrained(base_model_reload, model_name_upload)
14tokenizer_reload = AutoTokenizer.from_pretrained(model_name_upload)
15
16# GPU로 이동
17device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
18model_reload = model_reload.to(device)
19model_reload.eval()
20
21print("모델 로드 완료!")
22print("이제 이 코드로 어디서든 내 모델을 사용할 수 있습니다!")
23
24**Note**: 이 모델은 교육 목적으로 만들어졌습니다.