Kronfluence bug

[dilyabareeva]

Describe the bug
When running the tests on cuda, I get the error:

---
>           raise NotImplementedError(error_msg)
E           NotImplementedError: Automatic batch size search is not supported for multi-GPU setting. Please manually configure the batch size by passing in `per_device_batch_size`.

To Reproduce
Steps to reproduce the behavior:

1. Run the tests on cuda

Expected behavior
Tests don't fail.

Desktop (please complete the following information):

• OS: Linux
• Python 3.11