sync with working jax inference code from main repo e0cb68e Contrebande Labs commited on Apr 28, 2023
changed CUDA version for compatibility with the HF spaces 2b3e373 Contrebande Labs commited on Apr 28, 2023