eadadi

vim_ai_complete_default and vim_ai_edit_default model can't be changed

10

When changing the model of either vim_ai_complete_default or vim_ai_edit_default to another one (e.g. gpt-4o) then they stop working, and an error suggesting that API key should be verified is logged.

atari100k:atari_pong fails on Batcher creation

4

Running with: ```bash python3 ./recall2imagine/train.py / --logdir /logdir/$(date +%Y%m%d-%H%M%S) / --configs atari100k tiny --task atari_pong --jax.platform=cpu ``` Fails with: ``` AttributeError: 'function' object has no attribute 'serializer' ``` Traceback: ```...