KeyError: 'state_dict'

[zhuofengli]

I hit the following stack trace
Traceback (most recent call last): File "/Users/zhuofengli/Ckpt2Diff/ckpt2diff.py", line 154, in <module> wizard() File "/usr/local/lib/python3.10/site-packages/click/core.py", line 1130, in __call__ return self.main(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/click/core.py", line 1055, in main rv = self.invoke(ctx) File "/usr/local/lib/python3.10/site-packages/click/core.py", line 1404, in invoke return ctx.invoke(self.callback, **ctx.params) File "/usr/local/lib/python3.10/site-packages/click/core.py", line 760, in invoke return __callback(*args, **kwargs) File "/Users/zhuofengli/Ckpt2Diff/ckpt2diff.py", line 148, in wizard convert(converter, CKPT_FILE, load_config(CONFIG_FILE, DEFAULT_CONFIG_URL), scheduler, ema, HF_MODEL_DIR) File "/Users/zhuofengli/Ckpt2Diff/ckpt2diff.py", line 89, in convert convert_full_checkpoint( File "<string>", line 806, in convert_full_checkpoint KeyError: 'state_dict'
any way to get around it?