Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -32,7 +32,18 @@ language:
|
|
| 32 |
|
| 33 |
## 사용 방법
|
| 34 |
|
| 35 |
-
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 36 |
```
|
| 37 |
|
| 38 |
## 모델 정보
|
|
|
|
| 32 |
|
| 33 |
## 사용 방법
|
| 34 |
|
| 35 |
+
```# 1. 토크나이저 불러오기
|
| 36 |
+
tokenizer_reload = AutoTokenizer.from_pretrained(model_name_upload)
|
| 37 |
+
|
| 38 |
+
# 2. 베이스 모델 불러오기
|
| 39 |
+
base_reload = AutoModelForCausalLM.from_pretrained("skt/kogpt2-base-v2")
|
| 40 |
+
|
| 41 |
+
# 3. 학습할 때와 똑같이 크기 조정
|
| 42 |
+
base_reload.resize_token_embeddings(len(tokenizer_reload))
|
| 43 |
+
base_reload.config.pad_token_id = tokenizer_reload.pad_token_id
|
| 44 |
+
|
| 45 |
+
# 4. LoRA 불러오기
|
| 46 |
+
model_reload = PeftModel.from_pretrained(base_reload, model_name_upload)
|
| 47 |
```
|
| 48 |
|
| 49 |
## 모델 정보
|