Update README.md
Browse files
README.md
CHANGED
@@ -12,8 +12,8 @@ tags:
|
|
12 |
```
|
13 |
from transforemrs import AutoTokenizer, AutoModelForCausalLM
|
14 |
|
15 |
-
model = AutoModelForCausalLM.from_pretrained("AIdenU/Gemma-7b-
|
16 |
-
tokenizer = AutoTokenizer.from_pretrained("AIdenU/Gemma-7b-
|
17 |
|
18 |
systemPrompt = "๋น์ ์ ์ ๋ฅํ AI์
๋๋ค."
|
19 |
prompt = "์ง๋ ์ด๋ ๋ฐ์ผ๋ฉด ๊ฟํํ๋์?"
|
|
|
12 |
```
|
13 |
from transforemrs import AutoTokenizer, AutoModelForCausalLM
|
14 |
|
15 |
+
model = AutoModelForCausalLM.from_pretrained("AIdenU/Gemma-7b-ko-Y24_v2.0", device_map="auto")
|
16 |
+
tokenizer = AutoTokenizer.from_pretrained("AIdenU/Gemma-7b-ko-Y24_v2.0", use_fast=True)
|
17 |
|
18 |
systemPrompt = "๋น์ ์ ์ ๋ฅํ AI์
๋๋ค."
|
19 |
prompt = "์ง๋ ์ด๋ ๋ฐ์ผ๋ฉด ๊ฟํํ๋์?"
|