Error 'global_step'

#4
by radames HF staff - opened

Hi @sweetcocoa very cool project! currently trying to send you a PR to add cached examples to your demo.
however running in both colab and here (after duplicating your space) I'm running on this error.
Has anything changed on your model repo?
thanks

Traceback (most recent call last):
  File "app.py", line 28, in <module>
    wrapper, model_id, config = model_load()
  File "app.py", line 15, in model_load
    wrapper = wrapper.load_from_checkpoint(
  File "/home/user/.pyenv/versions/3.8.9/lib/python3.8/site-packages/pytorch_lightning/core/saving.py", line 139, in load_from_checkpoint
    return _load_from_checkpoint(
  File "/home/user/.pyenv/versions/3.8.9/lib/python3.8/site-packages/pytorch_lightning/core/saving.py", line 163, in _load_from_checkpoint
    checkpoint = _pl_migrate_checkpoint(
  File "/home/user/.pyenv/versions/3.8.9/lib/python3.8/site-packages/pytorch_lightning/utilities/migration/utils.py", line 114, in _pl_migrate_checkpoint
    checkpoint, migrations = migrate_checkpoint(checkpoint)
  File "/home/user/.pyenv/versions/3.8.9/lib/python3.8/site-packages/pytorch_lightning/utilities/migration/utils.py", line 62, in migrate_checkpoint
    checkpoint = migration_function(checkpoint)
  File "/home/user/.pyenv/versions/3.8.9/lib/python3.8/site-packages/pytorch_lightning/utilities/migration/migration.py", line 85, in _migrate_loop_global_step_to_progress_tracking
    global_step = checkpoint["global_step"]
KeyError: 'global_step'

Same error. Please help!

Same error! is it fixed yet?

hi @susnato , @Rodrix3333 found the issue,
please pin the versions following this

pretty-midi==0.2.9
omegaconf==2.1.1
transformers==4.16.1 
pytorch-lightning==1.8.3
essentia==2.1b6.dev609 
note-seq==0.0.3 
pyFluidSynth==1.3.0
--extra-index-url https://download.pytorch.org/whl/cpu
torch==1.13.1
torchaudio==0.13.1

PR to fix it

radames changed discussion status to closed

Sign up or log in to comment