Update README.md
Browse files
README.md
CHANGED
@@ -246,6 +246,21 @@ print(question, response)
|
|
246 |
# 2. 熊猫的皮毛颜色和图案不同:红熊猫的皮毛是橙红色,脸部、耳朵和四肢的末端有白色斑块;而大熊猫的皮毛是黑白相间的,眼睛、耳朵和四肢的末端是黑色的,脸部、耳朵内侧和身体其他部分是白色的。
|
247 |
# 3. 熊猫的姿态和位置不同:红熊猫坐在一个木制的结构上,而大熊猫坐在地上。
|
248 |
# 4. 背景中的植被和环境细节略有不同,但都包含树木和竹子。
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
249 |
```
|
250 |
|
251 |
## Citation
|
|
|
246 |
# 2. 熊猫的皮毛颜色和图案不同:红熊猫的皮毛是橙红色,脸部、耳朵和四肢的末端有白色斑块;而大熊猫的皮毛是黑白相间的,眼睛、耳朵和四肢的末端是黑色的,脸部、耳朵内侧和身体其他部分是白色的。
|
247 |
# 3. 熊猫的姿态和位置不同:红熊猫坐在一个木制的结构上,而大熊猫坐在地上。
|
248 |
# 4. 背景中的植被和环境细节略有不同,但都包含树木和竹子。
|
249 |
+
|
250 |
+
# batch inference
|
251 |
+
pixel_values1 = load_image('./examples/image1.jpg', max_num=6).to(torch.bfloat16).cuda()
|
252 |
+
pixel_values2 = load_image('./examples/image2.jpg', max_num=6).to(torch.bfloat16).cuda()
|
253 |
+
image_counts = [pixel_values1.size(0), pixel_values2.size(0)]
|
254 |
+
pixel_values = torch.cat((pixel_values1, pixel_values2), dim=0)
|
255 |
+
|
256 |
+
questions = ["Describe the image in detail."] * len(image_counts)
|
257 |
+
responses = model.batch_chat(tokenizer, pixel_values,
|
258 |
+
image_counts=image_counts,
|
259 |
+
questions=questions,
|
260 |
+
generation_config=generation_config)
|
261 |
+
for question, response in zip(questions, responses):
|
262 |
+
print(question)
|
263 |
+
print(response)
|
264 |
```
|
265 |
|
266 |
## Citation
|