Skip to content

Commit f66fd96

Browse files
committed
Patch llava
1 parent 3daa2ba commit f66fd96

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/models/llava/runner/llava_image_prefiller.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,8 +34,8 @@ class ET_EXPERIMENTAL LlavaImagePrefiller {
3434
::executorch::extension::llm::Image& image,
3535
int64_t& start_pos) {
3636
auto image_tensor = executorch::extension::from_blob(
37-
image.data.data(),
38-
{3, image.height, image.width},
37+
image.get_uint8_data().data(),
38+
{3, image.height(), image.width()},
3939
::executorch::aten::ScalarType::Byte);
4040
// Run image encoder
4141
auto image_encoder_outputs =

0 commit comments

Comments
 (0)