search

JeffSHF / ColabDock

Code for ColabDock paper
Other
114 stars 12 forks source link

'Config' object has no attribute 'jax_experimental_name_stack' #8

Closed VGPReys closed 1 year ago

VGPReys commented 1 year ago

I do have the same issue as the one in https://github.com/JeffSHF/ColabDock/issues/3.

By running the example python main.py, I get the 'Config' object has no attribute 'jax_experimental_name_stack' error from jax.

Python 3.8.17 (default, Jul 5 2023, 21:04:15) [GCC 11.2.0] :: Anaconda, Inc. on linux Type "help", "copyright", "credits" or "license" for more information.

import jax
jax.__version__

'0.4.13'

How should I solve this ?

JeffSHF commented 1 year ago

Hi, can you update haiku to 0.0.10 and have another try?

VGPReys commented 1 year ago

Hi, Indeed, it solved my problem ! Thanks a lot.