jangmin commited on
Commit
268c9c7
โ€ข
1 Parent(s): 42155d0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -3
README.md CHANGED
@@ -43,6 +43,7 @@ The current model was developed using the GPT-4 API to generate a dataset for or
43
  ## How to Get Started with the Model
44
 
45
  This is a simple example of usage of the model.
 
46
 
47
  ``` python
48
  import torch
@@ -88,9 +89,10 @@ tokenizer = AutoTokenizer.from_pretrained(
88
 
89
  sentence = "์•„์ด์Šค์•„๋ฉ”๋ฆฌ์นด๋…ธ ํ†จ์‚ฌ์ด์ฆˆ ํ•œ์ž” ํ•˜๊ณ ์š”. ๋”ธ๊ธฐ์Šค๋ฌด๋”” ํ•œ์ž” ์ฃผ์„ธ์š”. ๋˜, ์ฝœ๋“œ๋ธŒ๋ฃจ๋ผ๋–ผ ํ•˜๋‚˜์š”."
90
  analysis = wrapper_generate(
91
- model=trained_model,
92
- tokenizer=tokenier,
93
- input_prompt=prompt_templateformat(System=default_system_msg, User=sentece, do_stream=True)
 
94
  )
95
  print(analysis)
96
  ```
 
43
  ## How to Get Started with the Model
44
 
45
  This is a simple example of usage of the model.
46
+ If you want to load the fined-tuned model in INT4, please specify @load_in_4bit=True@ instead of @load_in_8bit=True@.
47
 
48
  ``` python
49
  import torch
 
89
 
90
  sentence = "์•„์ด์Šค์•„๋ฉ”๋ฆฌ์นด๋…ธ ํ†จ์‚ฌ์ด์ฆˆ ํ•œ์ž” ํ•˜๊ณ ์š”. ๋”ธ๊ธฐ์Šค๋ฌด๋”” ํ•œ์ž” ์ฃผ์„ธ์š”. ๋˜, ์ฝœ๋“œ๋ธŒ๋ฃจ๋ผ๋–ผ ํ•˜๋‚˜์š”."
91
  analysis = wrapper_generate(
92
+ model=trained_model,
93
+ tokenizer=tokenizer,
94
+ input_prompt=prompt_template.format(System=default_system_msg, User=sentence),
95
+ do_stream=False
96
  )
97
  print(analysis)
98
  ```