TypeError: Expected state_dict to be dict-like, got <class 'lit_llama.utils.lazy_load'>.

[ChangeNext]

Hi

Thanks for your work. I'm encountering an issue when I attempt to run eval.py

eval.py line 112 model.load_state_dict(pretrained_checkpoint, strict=False)

TypeError: Expected state_dict to be dict-like, got <class 'lit_llama.utils.lazy_load'>.