dalle-mini icon indicating copy to clipboard operation
dalle-mini copied to clipboard

Published 20 hours ago • verified publisher icon borisdayma

fix: adapt to new jax API

Open GaetanLepage opened this issue 9 months ago • 2 comments

Tests are currently broken when ran with the latest version of JAX installed. This patch ensures the code compatibility with the current jax API.

GaetanLepage avatar Nov 09 '23 09:11 GaetanLepage

Thanks, that's all you needed to change? I'm scared the training code will most likely be broken with latest version as well.

I pinned the versions to use here: https://github.com/borisdayma/dalle-mini/blob/f0be4de610285a002052024a1e096126f9452cc4/setup.cfg#L32

borisdayma avatar Nov 15 '23 15:11 borisdayma

Thanks, that's all you needed to change?

Well at least, it is enough for all of the tests to pass. For the context, I was updating dalle-mini to the latest version on nixpkgs.

GaetanLepage avatar Nov 15 '23 22:11 GaetanLepage