Function create_data_loader

internvl_chat_llava/llava/eval/model_vqa_loader.py:65–69

(questions, image_folder, tokenizer, image_processor, model_config, batch_size=1, num_workers=4)

# DataLoader
def create_data_loader(questions, image_folder, tokenizer, image_processor, model_config, batch_size=1, num_workers=4):
    assert batch_size == 1, "batch_size must be 1"
    dataset = CustomDataset(questions, image_folder, tokenizer, image_processor, model_config)
    data_loader = DataLoader(dataset, batch_size=batch_size, num_workers=num_workers, shuffle=False)
    return data_loader


def eval_model(args):