Skip to content

Bug: image_grid_thw[image_index][0] IndexError: list index out of range #423

@luyouqi233

Description

@luyouqi233

是否应该修改roll/distributed/scheduler/generate_scheduler.py

gen_batch = request_data.pop(
    batch_keys=["input_ids", "attention_mask", "position_ids"], 
    non_tensor_batch_keys=["multi_modal_data"] if "multi_modal_data" in request_data.non_tensor_batch else []
)
gen_batch.meta_info = request_data.meta_info
num_return_sequences = generation_config["num_return_sequences"]
is_num_return_sequences_expand = self.pipeline_config.is_num_return_sequences_expand
if is_num_return_sequences_expand:
    request_data = request_data.repeat(repeat_times=num_return_sequences)
    gen_batch.meta_info["generation_config"]["num_return_sequences"] = 1
    num_return_sequences = 1

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions