Patrick von Platen
|
c39a511b5f
|
[Loading] Ignore unneeded files (#1107)
* [Loading] Ignore unneeded files
* up
|
2022-11-02 19:20:42 +01:00 |
|
Anton Lozhkov
|
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 |
|
Anton Lozhkov
|
2c82e0c4eb
|
Reorganize pipeline tests (#963)
* Reorganize pipeline tests
* fix vq
|
2022-10-24 16:34:01 +02:00 |
|
Patrick von Platen
|
1d51224403
|
[Flax] Complete tests (#828)
|
2022-10-13 18:18:32 +02:00 |
|
Patrick von Platen
|
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 |
|
Patrick von Platen
|
f1d4289be8
|
[Flax] Add test (#824)
|
2022-10-13 13:55:39 +02:00 |