UICHEOL-HWANG commited on
Commit
8eef067
·
verified ·
1 Parent(s): f5106f6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -1
README.md CHANGED
@@ -28,9 +28,14 @@ This llama model was trained 2x faster with [Unsloth](https://github.com/unsloth
28
 
29
  ```python
30
  from transformers import AutoTokenizer, AutoModelForCausalLM
 
31
 
32
  tokenizer = AutoTokenizer.from_pretrained("UICHEOL-HWANG/EcomGen-Llama3.2-3B")
33
- model = AutoModelForCausalLM.from_pretrained("UICHEOL-HWANG/EcomGen-Llama3.2-3B")
 
 
 
 
34
 
35
  instruction = """
36
  상품명: 프리미엄 유기농 쌀 10kg
 
28
 
29
  ```python
30
  from transformers import AutoTokenizer, AutoModelForCausalLM
31
+ import torch
32
 
33
  tokenizer = AutoTokenizer.from_pretrained("UICHEOL-HWANG/EcomGen-Llama3.2-3B")
34
+ model = AutoModelForCausalLM.from_pretrained(
35
+ "UICHEOL-HWANG/EcomGen-Llama3.2-3B",
36
+ torch_dtype=torch.bfloat16,
37
+ device_map="auto",
38
+ )
39
 
40
  instruction = """
41
  상품명: 프리미엄 유기농 쌀 10kg