File tree Expand file tree Collapse file tree 1 file changed +5
-2
lines changed
torchtune/datasets/multimodal Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -51,8 +51,11 @@ def vqa_dataset(
5151 in the filepath in ``data_files``, and set ``split="train"``. See `Hugging Face's
5252 <https://huggingface.co/docs/datasets/en/package_reference/loading_methods#datasets.load_dataset.path>`_
5353 ``load_dataset`` for more details.
54- image_dir (str): path to the directory containing the images as you are expected to download the COCO dataset
55- before using. Default is "coco/".
54+ image_dir (str): path to the directory containing the images that is prepended to all image
55+ paths in the dataset. For example, if ``image_dir="/home/user/dataset/"` and the sample image path
56+ was ``"images/1.jpg"``, the final image path that will be loaded is ``"/home/user/dataset/images/1.jpg"``.
57+ If None, assume images are available in current working directory or are located
58+ on a remote url. For text-only, leave as None. Default is None.
5659 column_map (Optional[Dict[str, str]]): a mapping to change the expected "input",
5760 "output", and "image" column names to the actual column names in the dataset. Keys should be "input",
5861 "output", and "image, and values should be the actual column names.
You can’t perform that action at this time.
0 commit comments