heegyu commited on
Commit
3433fd2
โ€ข
1 Parent(s): e83ba4b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -10,7 +10,7 @@ language:
10
 
11
  - Base Model: [42dot/42dot_LLM-SFT-1.3B](https://huggingface.co/42dot/42dot_LLM-SFT-1.3B)
12
  - [v0.1](https://huggingface.co/heegyu/ko-reward-model-1.3b-v0.1) ๋ชจ๋ธ์€ helpful + safety๋ฅผ ๊ฐ™์ด ํ•™์Šตํ–ˆ๊ณ  safeํ•œ ๋‹ต๋ณ€์— ์ง€๋‚˜์น˜๊ฒŒ ๋†’์€ ์ ์ˆ˜๋ฅผ ์ฃผ๋Š” ๊ฒฝํ–ฅ์ด ์žˆ์–ด์„œ ๋ถ„๋ฆฌ ํ›„ ๋”ฐ๋กœ ํ•™์Šตํ–ˆ์Šต๋‹ˆ๋‹ค.
13
- - ์ด ๋ชจ๋ธ์€ ์œค๋ฆฌ์ ์ธ ๋‹ต๋ณ€์— ๋†’์€ ์ ์ˆ˜๋ฅผ ์ฃผ๋Š” safety ๋ชจ๋ธ์ž…๋‹ˆ๋‹ค.
14
 
15
  ## Hyperparameters:
16
  - Batch: 128
@@ -36,7 +36,7 @@ language:
36
  ```
37
  from transformers import pipeline
38
 
39
- pipe = pipeline("text-classification", model="heegyu/1231-korm-safety-1.3b-128-1e-5")
40
 
41
 
42
  pipe("""<human>:
 
10
 
11
  - Base Model: [42dot/42dot_LLM-SFT-1.3B](https://huggingface.co/42dot/42dot_LLM-SFT-1.3B)
12
  - [v0.1](https://huggingface.co/heegyu/ko-reward-model-1.3b-v0.1) ๋ชจ๋ธ์€ helpful + safety๋ฅผ ๊ฐ™์ด ํ•™์Šตํ–ˆ๊ณ  safeํ•œ ๋‹ต๋ณ€์— ์ง€๋‚˜์น˜๊ฒŒ ๋†’์€ ์ ์ˆ˜๋ฅผ ์ฃผ๋Š” ๊ฒฝํ–ฅ์ด ์žˆ์–ด์„œ ๋ถ„๋ฆฌ ํ›„ ๋”ฐ๋กœ ํ•™์Šตํ–ˆ์Šต๋‹ˆ๋‹ค.
13
+ - ์ด ๋ชจ๋ธ์€ ์œค๋ฆฌ์ ์ธ ๋‹ต๋ณ€์— ๋†’์€ ์ ์ˆ˜๋ฅผ ์ฃผ๋Š” safety ๋ชจ๋ธ์ž…๋‹ˆ๋‹ค. ์œ ์šฉํ•˜๊ณ  ์ž์„ธํ•œ ๋‹ต๋ณ€์— ๋Œ€ํ•ด ๋†’์€ ์ ์ˆ˜๋ฅผ ์ฃผ๋Š” helpful ๋ชจ๋ธ์€ [heegyu/ko-reward-model-helpful-1.3b-v0.2](https://huggingface.co/heegyu/ko-reward-model-helpful-1.3b-v0.2) <- ์ด ๋ชจ๋ธ์„ ์‚ฌ์šฉํ•˜์„ธ์š”
14
 
15
  ## Hyperparameters:
16
  - Batch: 128
 
36
  ```
37
  from transformers import pipeline
38
 
39
+ pipe = pipeline("text-classification", model="heegyu/ko-reward-model-safety-1.3b-v0.2")
40
 
41
 
42
  pipe("""<human>: