INVOCATION FAILED partially initialized module 'jax' has no attribute 'version'

[zladic]

INVOCATION FAILED
partially initialized module 'jax' has no attribute 'version' (most likely due to a circular import)

still getting this error

also
METADATA ERROR
This model's metadata could not be read. It may still work, but it's possible the file is corrupt or otherwise unsupported. The error was: AttributeError(partially initialized module 'jax' has no attribute 'version' (most likely due to a circular import))
fo the checkpoint choosen
any adv?

Thx!
BR