ssui-liu commited on
Commit
4226e3e
1 Parent(s): 646cf59

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -1
README.md CHANGED
@@ -5,6 +5,10 @@ datasets:
5
  language:
6
  - en
7
  ---
 
 
 
 
8
  from peft import PeftModel, PeftConfig
9
  from transformers import AutoModelForCausalLM, AutoTokenizer
10
 
@@ -37,4 +41,5 @@ response = tokenizer.decode(response[0],
37
  skip_special_tokens=False,
38
  clean_up_tokenization_spaces=False
39
  )
40
- print(response)
 
 
5
  language:
6
  - en
7
  ---
8
+
9
+ ## Inference Example:
10
+
11
+ ```python
12
  from peft import PeftModel, PeftConfig
13
  from transformers import AutoModelForCausalLM, AutoTokenizer
14
 
 
41
  skip_special_tokens=False,
42
  clean_up_tokenization_spaces=False
43
  )
44
+ print(response)
45
+ ```