Update README.md
Browse files
README.md
CHANGED
@@ -18,8 +18,8 @@ tags:
|
|
18 |
|
19 |
```
|
20 |
SYSTEM_PROMPT = "### System:\n ๋น์ ์ ์ฐจ๊ทผ์ฐจ๊ทผ ์๊ฐํ๊ณ , ๋
ผ๋ฆฌ์ ์ธ ์ฌ๊ณ ๋ฅผ ํตํด ์ฃผ์ด์ง ๋ฌธ์ ๋ฅผ ํธ๋ ์ธ๊ณต์ง๋ฅ ์
๋๋ค."
|
21 |
-
USER_PROMPT = "
|
22 |
-
ASSISTANT_PROMPT = "
|
23 |
template = f"{SYSTEM_PROMPT}{USER_PROMPT}{ASSISTANT_PROMPT}"
|
24 |
|
25 |
inputs = tokenizer(template, retuen_tensors="pt").to(0) # ๋ชจ๋ธ๊ณผ ๋์ผํ GPU์ ์ฌ์ฉํด์ผํจ. ์์ผ๋ฉด to(0) ์ง์ฐ์ธ์.
|
|
|
18 |
|
19 |
```
|
20 |
SYSTEM_PROMPT = "### System:\n ๋น์ ์ ์ฐจ๊ทผ์ฐจ๊ทผ ์๊ฐํ๊ณ , ๋
ผ๋ฆฌ์ ์ธ ์ฌ๊ณ ๋ฅผ ํตํด ์ฃผ์ด์ง ๋ฌธ์ ๋ฅผ ํธ๋ ์ธ๊ณต์ง๋ฅ ์
๋๋ค."
|
21 |
+
USER_PROMPT = "\n ### User: \n ๋ฏผ์์ด๋ ์ด์ ์ฌ๊ณผ๋ฅผ 3๊ฐ ๋ค๊ณ ์์์ต๋๋ค. ์ค๋ ๋ฏผ์์ด๊ฐ ์ฌ๊ณผ๋ฅผ ๋ฌ๋ผ๊ณ ํ๋๋ฐ ์ฃผ์ง ์์์ต๋๋ค. ๋ฏผ์์ด๋ ๋์ ์ฌ๋์
๋๊น?"
|
22 |
+
ASSISTANT_PROMPT = "\n ### Assistant: \n"
|
23 |
template = f"{SYSTEM_PROMPT}{USER_PROMPT}{ASSISTANT_PROMPT}"
|
24 |
|
25 |
inputs = tokenizer(template, retuen_tensors="pt").to(0) # ๋ชจ๋ธ๊ณผ ๋์ผํ GPU์ ์ฌ์ฉํด์ผํจ. ์์ผ๋ฉด to(0) ์ง์ฐ์ธ์.
|