Update README.md
Browse files
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](
|
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
|
|
|
|
|
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 |
```
|