GitLab

* finish

* fix

* Update tests/models/test_models_unet_2d.py

* style
Co-authored-by: Anton Lozhkov <anton@huggingface.co>