a7ca03aa85
Replace flake8 with ruff and update black ( #2279 )
...
* before running make style
* remove left overs from flake8
* finish
* make fix-copies
* final fix
* more fixes
2023-02-07 23:46:23 +01:00
e0e86b7470
Make height and width optional ( #1401 )
...
* fix
* add test
* fix test
* uP
* up
* fix some tests
2022-11-24 18:23:59 +01:00
af279434d0
Flax tests: don't hardcode number of devices ( #1175 )
...
Flax tests: don't hardcode number of devices.
This makes it possible to test on CPU/GPU. However, expected slices are
only checked when there are 8 devices.
2022-11-09 20:04:43 +01:00
c39a511b5f
[Loading] Ignore unneeded files ( #1107 )
...
* [Loading] Ignore unneeded files
* up
2022-11-02 19:20:42 +01:00
4e59bcc680
[CI] Framework and hardware-specific CI tests ( #997 )
...
* [WIP][CI] Framework and hardware-specific docker images for CI tests
* username
* fix cpu
* try out the image
* push latest
* update workspace
* no root isolation for actions
* add a flax image
* flax and onnx matrix
* fix runners
* add reports
* onnxruntime image
* retry tpu
* fix
* fix
* build onnxruntime
* naming
* onnxruntime-gpu image
* onnxruntime-gpu image, slow tests
* latest jax version
* trigger flax
* run flax tests in one thread
* fast flax tests on cpu
* fast flax tests on cpu
* trigger slow tests
* rebuild torch cuda
* force cuda provider
* fix onnxruntime tests
* trigger slow
* don't specify gpu for tpu
* optimize
* memory limit
* fix flax tests
* disable docker cache
2022-11-02 14:07:07 +01:00
2c82e0c4eb
Reorganize pipeline tests ( #963 )
...
* Reorganize pipeline tests
* fix vq
2022-10-24 16:34:01 +02:00
1d51224403
[Flax] Complete tests ( #828 )
2022-10-13 18:18:32 +02:00
7c2262640b
Align PT and Flax API - allow loading checkpoint from PyTorch configs ( #827 )
...
* up
* finish
* add more tests
* up
* up
* finish
2022-10-13 17:43:06 +02:00
f1d4289be8
[Flax] Add test ( #824 )
2022-10-13 13:55:39 +02:00
Patrick von Platen