PerRing commited on
Commit
0257105
1 Parent(s): 09b9df3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -12,8 +12,9 @@ pip install --upgrade accelerate fbgemm-gpu torch
12
  ```python
13
  from transformers import AutoTokenizer, AutoModelForCausalLM
14
 
15
- tokenizer = AutoTokenizer.from_pretrained("PerRing/llama-3-Korean-Bllossom-8B-fp8")
16
- model = AutoModelForCausalLM.from_pretrained("PerRing/llama-3-Korean-Bllossom-8B-fp8",device_map='auto')
 
17
  ```
18
 
19
  ```python
 
12
  ```python
13
  from transformers import AutoTokenizer, AutoModelForCausalLM
14
 
15
+ model_id="PerRing/llama-3-Korean-Bllossom-8B-fp8"
16
+ tokenizer = AutoTokenizer.from_pretrained(model_id)
17
+ model = AutoModelForCausalLM.from_pretrained(model_id,device_map='auto')
18
  ```
19
 
20
  ```python