Unable to follow the finetune notebook
nkn002 opened this issue · 0 comments
nkn002 commented
Hi,
I was trying to follow the notebook but I couldn't run it for the summarization task. When I tried following it, I get the following error:
from jax.experimental.global_device_array import GlobalDeviceArray ModuleNotFoundError: No module named 'jax.experimental.global_device_array'
Is there a way to fix this issue?