Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 9 additions & 3 deletions tests/models/minicpmv4_6/test_modeling_minicpmv4_6.py
Original file line number Diff line number Diff line change
Expand Up @@ -497,9 +497,11 @@ def test_small_model_vision_generation_batch(self):
expected_texts = Expectations(
{
("cuda", None): [
"The animal in the image is a Pystylus, also known as the Eurasian pystylus or snow leopard cat. It's a",
"The animal in the image is a Pystylus, also known as the Eurasian pystylus or snow leopard cat. It's a",
],
"The animal in the image is a Pystylus, also known as a Pystylus cat or Eurasian pystylus. It",
] * 2,
("cuda", 10): [
"The animal in the image is a Pystylus, also known as a Eurasian pystylus or snow leopard cat. It's a",
] * 2,
}
) # fmt: skip
EXPECTED_TEXT = expected_texts.get_expectation()
Expand Down Expand Up @@ -547,6 +549,10 @@ def test_small_model_vision_generation_batch_mixed(self):
"The animal in the image is a Pystylus, also known as the Eurasian pystylus or snow leopard cat. It's a",
"I'm a model from the MiniCPM series, developed by Modelbest and OpenBMB. For more details, you can visit https://github",
],
("cuda", 10): [
"The animal in the image is a Pystylus, also known as a Eurasian pystylus or snow leopard cat. It's a",
"I'm a model from the MiniCPM series, developed by Modelbest and OpenBMB. For more details, you can visit https://github",
],
}
) # fmt: skip
EXPECTED_TEXT = expected_texts.get_expectation()
Expand Down
Loading