Please help me

[Deerzh]

When I run this command: python3 inference.py --test_dir music/demo_song --output_dir outputs/demo_song --model checkpoints/epoch_1.pt --dance_num 6, an error occurred: Traceback (most recent call last): File "inference.py", line 18, in from v2.generator import Generator File "/home/zhang/compatibility_analysis/DanceRevolution/v2/generator.py", line 8, in from model import Encoder, Decoder, Model, get_subsequent_mask ModuleNotFoundError: No module named 'model'

Can you help me fix this issue?