StatPan commited on
Commit
6b23c57
โ€ข
1 Parent(s): 4062944

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -18,8 +18,8 @@ tags:
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) ์ง€์šฐ์„ธ์š”.
 
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) ์ง€์šฐ์„ธ์š”.