jmoso13 / jukebox-diffusion Goto Github PK

--seconds-length 8 --context-audio /content/shadow_synth_441.wav --save-dir /content/drive/MyDrive/AI/Jbdiff --project-name jbdiff-fun --levels 12 --noise-style constant --noise-step-size 0.7 --dd-noise-style walk --dd-noise-step-size 0.15 --token-multiplier 4 --update-lowest-context False 

Using cuda True
levels:  [1, 2]
Loading Level 2
Downloading from azure
Traceback (most recent call last):
  File "/content/jukebox-diffusion/sample.py", line 269, in <module>
    main()
  File "/content/jukebox-diffusion/sample.py", line 262, in main
    run(**vars(args))
  File "/content/jukebox-diffusion/sample.py", line 94, in run
    vqvae, _, _ = make_jb(".", level, batch_size, base_tokens, context_mult, aug_shift, num_workers=12, train=False)
  File "/content/jukebox-diffusion/jbdiff/utils.py", line 186, in make_jb
    vqvae = make_vqvae(hps, device)
  File "/usr/local/lib/python3.10/dist-packages/jukebox/make_models.py", line 95, in make_vqvae
    restore_model(hps, vqvae, hps.restore_vqvae)
  File "/usr/local/lib/python3.10/dist-packages/jukebox/make_models.py", line 55, in restore_model
    checkpoint = load_checkpoint(checkpoint_path)
  File "/usr/local/lib/python3.10/dist-packages/jukebox/make_models.py", line 37, in load_checkpoint
    checkpoint = t.load(restore, map_location=t.device('cpu'))
  File "/usr/local/lib/python3.10/dist-packages/torch/serialization.py", line 1028, in load
    return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
  File "/usr/local/lib/python3.10/dist-packages/torch/serialization.py", line 1246, in _legacy_load
    magic_number = pickle_module.load(f, **pickle_load_args)
EOFError: Ran out of input