Skip to content

Commit ea5b906

Browse files
authored
Fix typo in multimodal_datasets.rst
1 parent b7e4838 commit ea5b906

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docs/source/basics/multimodal_datasets.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ in the text, ``"<image>"`` for where to place the image tokens. This will get re
6464
split="train",
6565
)
6666
tokenized_dict = ds[0]
67-
print(transform.decode(tokenized_dict["tokens"], skip_special_tokens=False))
67+
print(model_transform.decode(tokenized_dict["tokens"], skip_special_tokens=False))
6868
# '<|begin_of_text|><|start_header_id|>user<|end_header_id|>\n\nQuestion:<|image|>What time is it on the clock?Answer:<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\nIt is 10:00AM.<|eot_id|>'
6969
print(tokenized_dict["encoder_input"]["images"][0].shape) # (num_tiles, num_channels, tile_height, tile_width)
7070
# torch.Size([4, 3, 224, 224])

0 commit comments

Comments
 (0)