skwh54 commited on
Commit
b9e3ca2
·
verified ·
1 Parent(s): 30229c0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -2
README.md CHANGED
@@ -4,7 +4,7 @@ base_model:
4
  - google/gemma-2-2b-it
5
  ---
6
 
7
- # Gemma-2-2b-it 초등학생 글 변환기
8
 
9
 
10
  ## Model Description
@@ -13,7 +13,16 @@ base_model:
13
 
14
  ## Usage
15
  ```
16
- print("hello")
 
 
 
 
 
 
 
 
 
17
  ```
18
 
19
  ## Training data
 
4
  - google/gemma-2-2b-it
5
  ---
6
 
7
+ # Gemma-2-2b 초등학생 글 변환기
8
 
9
 
10
  ## Model Description
 
13
 
14
  ## Usage
15
  ```
16
+ import transformers
17
+ import torch
18
+ from huggingface_hub import notebook_login
19
+
20
+ notebook_login()
21
+
22
+ BASE_MODEL = "skwh54/Gemma-2-2b-it-elementary-style-document"
23
+
24
+ model = AutoModelForCausalLM.from_pretrained(BASE_MODEL, device_map={"":0})
25
+ tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL)
26
  ```
27
 
28
  ## Training data