Add batch dim idx to support latest deepspeed DistributedAttention #6633
fast_tests.yml
on: pull_request
Run tests for optimum.habana.transformers
2m 24s
Run tests for optimum.habana.diffusers
14m 20s