search

jaswon / osu-dreamer

a diffusion-based ML model for generating osu! maps from raw audio
MIT License
115 stars 10 forks source link

ImportError: No module named pytorch_lightning.cli #10

Open Sorok-Dva opened 1 year ago

Sorok-Dva commented 1 year ago

Describe the bug Hello, I got an error after the installation while trying to start Model training

To Reproduce Steps to reproduce the behavior:

  1. pip install ./osu-dreamer
  2. python scripts/cli.py fit -c config.yml -c osu_dreamer/model/model.yml --data.src_path /mnt/h/osu!/Songs

Expected behavior The python script to be executed

Stacktrace

Traceback (most recent call last):
  File "scripts/cli.py", line 1, in <module>
    from pytorch_lightning.cli import LightningCLI
ImportError: No module named pytorch_lightning.cli