indexSelectLargeIndex: Device-side assertion `srcIndex < srcSelectDimSize' failed.

[radna0]

Anyone having this problem?

___2____Input tokenized shape: torch.Size([1, 57])
/var/lib/jenkins/pytorch/aten/src/ATen/native/hip/Indexing.hip:1294: indexSelectLargeIndex: Device-side assertion `srcIndex < srcSelectDimSize' failed.

        if self.use_video:
            print("___1____Video shape:", video.size())
            video = self.visual_encoder(video)  # B T D
            if self.proj_v2t is not None:
                video = self.proj_v2t(video)
            atts_vis = torch.ones(video.size()[:-1], dtype=torch.long).to(video.device)
            print("___1____atts_vis shape:", atts_vis.size())

        if self.use_speech:
            print("___2____Input tokenized shape:", input_tokenized['input_ids'].size())
            text = self.t5_model.encoder.embed_tokens(input_tokenized['input_ids'])  # B L D
            print("___2____Text tokenized shape:", text)
            encoded = self.t5_model.encoder(
                attention_mask=input_tokenized['attention_mask'],
                inputs_embeds=text,
            )
            print("___2____Encoded shape:", encoded.last_hidden_state.size())