Fix `AlignModelTest` tests
# What does this PR do?

- Fix `AlignModelTest` torchscript tests.  This model has non-persistent buffer, and need a bit more care (same as in common test)
```python
        self.register_buffer(
            "token_type_ids", torch.zeros(self.position_ids.size(), dtype=torch.long), persistent=False
        )
```

- Fix `AlignModelTest.test_multi_gpu_data_parallel_forward`: same as CLIP, we need a even number for `batch_size`, as this model have `logits_per_image` and `logits_per_text` which is of shape `(batch_size, batch_size)`, and in order to gather across devices, we need the second dim being the same. 
_The documentation is not available anymore as the PR was closed or merged._