BAAI
/

BoyaWu10 commited on
Commit
5bc6c4f
β€’
1 Parent(s): 324fb01

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -2
README.md CHANGED
@@ -9,7 +9,7 @@ license: apache-2.0
9
  <img src="./icon.png" alt="Logo" width="350">
10
  </p>
11
 
12
- πŸ“– [Technical report](https://arxiv.org/abs/2402.11530) | 🏠 [Code](https://github.com/BAAI-DCAI/Bunny) | 🐰 [Demo](https://wisemodel.cn/spaces/baai/Bunny)
13
 
14
  This is the merged weights of [bunny-phi-2-siglip-lora](https://huggingface.co/BAAI/bunny-phi-2-siglip-lora).
15
 
@@ -76,7 +76,9 @@ output_ids = model.generate(
76
  input_ids,
77
  images=image_tensor,
78
  max_new_tokens=100,
79
- use_cache=True)[0]
 
 
80
 
81
  print(tokenizer.decode(output_ids[input_ids.shape[1]:], skip_special_tokens=True).strip())
82
  ```
 
9
  <img src="./icon.png" alt="Logo" width="350">
10
  </p>
11
 
12
+ πŸ“– [Technical report](https://arxiv.org/abs/2402.11530) | 🏠 [Code](https://github.com/BAAI-DCAI/Bunny) | 🐰 [Demo](http://bunny.dataoptim.org)
13
 
14
  This is the merged weights of [bunny-phi-2-siglip-lora](https://huggingface.co/BAAI/bunny-phi-2-siglip-lora).
15
 
 
76
  input_ids,
77
  images=image_tensor,
78
  max_new_tokens=100,
79
+ use_cache=True
80
+ repetition_penalty=1.0 # increase this to avoid chattering
81
+ )[0]
82
 
83
  print(tokenizer.decode(output_ids[input_ids.shape[1]:], skip_special_tokens=True).strip())
84
  ```