huihui-ai commited on
Commit
be41e22
·
verified ·
1 Parent(s): 8d52238

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -1
README.md CHANGED
@@ -57,7 +57,12 @@ inputs = processor.apply_chat_template(
57
 
58
  # autoregressively complete prompt
59
  output = model.generate(**inputs, max_new_tokens=100)
60
- print(processor.decode(output[0], skip_special_tokens=True))
 
 
 
 
 
61
  ```
62
 
63
  ### Donation
 
57
 
58
  # autoregressively complete prompt
59
  output = model.generate(**inputs, max_new_tokens=100)
60
+ # print(processor.decode(output[0], skip_special_tokens=True))
61
+
62
+ response = processor.decode(output[0], skip_special_tokens=True)
63
+ cleaned_response = response.split("<|assistant|>")[1].split("<")[0].strip()
64
+
65
+ print(cleaned_response)
66
  ```
67
 
68
  ### Donation