Jingyao Li

Hi @DrSleep, I also meet the problem when I run the inference.py. I retain the model using the following setup, because the original batch size is out of memory I...

I am running the same issue. My Jax version is 0.4.23 --------------------------------------------------------------------------- ImportError Traceback (most recent call last) Cell In[8], line 1 ----> 1 import scvi File ~/mambaforge/mambaforge/envs/scvi/lib/python3.9/site-packages/scvi/__init__.py:10 7 from...

It was fixed by install flax see #2216