minmingbb commited on
Commit
48dcaf5
·
verified ·
1 Parent(s): f24f79c

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +12 -1
README.md CHANGED
@@ -32,7 +32,18 @@ language:
32
 
33
  ## 사용 방법
34
 
35
- ```python
 
 
 
 
 
 
 
 
 
 
 
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
  ## 모델 정보