Accelerate launch command fails to run

[hjzhang-forward]

The following command fails to execute: accelerate launch --config_file configs/accelerate/deepspeed-8-gpu.yaml starvector/train/train.py config=configs/models/starvector-1b/im2svg-stack.yaml

Error: train.py

  loss = model(batch)

starvector_arch.py def forward(self, vision_embeds, input_ids, num_generations, attention_mask, num_logits_to_keep)

Here, batch has no input_ids, num_generations, attention_mask, num_logits_to_keep

[Yukinonooo]

Have you solved this problem? I also encountered the same problem.

[Eric-zhang-1111]

same! Have you solved this problem?