ModelCheckpoint() argument after ** must be a mapping, not ModelCheckpoint

[aifartist]

I tried to do inference but got the above error.
I'm using python3.10 on Ubuntu 22.04. i9-13900K + 4090
I fixed this by changing return cls(**checkpoint_dict) to return cls(**vars(checkpoint_dict))

  File "/home/dwood/LLM/yar/./venv/lib/python3.10/site-packages/scripts/train_project_gutenberg.py", line 95, in load
    return cls(**checkpoint_dict)
TypeError: __main__.ModelCheckpoint() argument after ** must be a mapping, not ModelCheckpoint