Description: When running the gdm_lab_4_1_attention_visualization.ipynb notebook, specifically during the installation of ai_foundations and its dependencies, an AttributeError occurs:
AttributeError: module 'kauldron.checkpoints' has no attribute 'AbstractPartialLoader'
This error arises because the ai_foundations library (installed from git+https://github.com/google-deepmind/ai-foundations.git@main) depends on the kauldron library, and the version of kauldron implicitly installed or available in the environment does not provide the AbstractPartialLoader attribute within its checkpoints module. The specific version kauldron==0.0.12 contains this attribute and resolves the error
Description: When running the gdm_lab_4_1_attention_visualization.ipynb notebook, specifically during the installation of ai_foundations and its dependencies, an AttributeError occurs:
AttributeError: module 'kauldron.checkpoints' has no attribute 'AbstractPartialLoader'
This error arises because the ai_foundations library (installed from git+https://github.com/google-deepmind/ai-foundations.git@main) depends on the kauldron library, and the version of kauldron implicitly installed or available in the environment does not provide the AbstractPartialLoader attribute within its checkpoints module. The specific version kauldron==0.0.12 contains this attribute and resolves the error