GuyTevet / motion-diffusion-model

The official PyTorch implementation of the paper "Human Motion Diffusion Model"
MIT License
3.12k stars 336 forks source link

AssertionError: Path ./body_models/smpl\SMPL_NEUTRAL.pkl does not exist! #4

Closed DrMemoryFish closed 2 years ago

DrMemoryFish commented 2 years ago

im getting the error:

(mdm) E:\AI\motion-diffusion-model>python -m sample --model_path ./save/humanml_trans_enc_512/model000200000.pt --input_text ./assets/example_text_prompts.txt
Creating model and diffusion...
TRANS_ENC init
EMBED TEXT
Loading CLIP...
100%|███████████████████████████████████████| 338M/338M [00:18<00:00, 18.8MiB/s]
Traceback (most recent call last):
  File "C:\Users\Abdullah\miniconda3\envs\mdm\lib\runpy.py", line 193, in _run_module_as_main
    "__main__", mod_spec)
  File "C:\Users\Abdullah\miniconda3\envs\mdm\lib\runpy.py", line 85, in _run_code
    exec(code, run_globals)
  File "E:\AI\motion-diffusion-model\sample.py", line 178, in <module>
    main()
  File "E:\AI\motion-diffusion-model\sample.py", line 45, in main
    model, diffusion = create_model_and_diffusion(args)
  File "E:\AI\motion-diffusion-model\utils\model_util.py", line 13, in create_model_and_diffusion
    model = MDM(**get_model_args(args))
  File "E:\AI\motion-diffusion-model\model\mdm.py", line 102, in __init__
    self.rot2xyz = Rotation2xyz(device='cpu', dataset=self.dataset)
  File "E:\AI\motion-diffusion-model\model\rotation2xyz.py", line 15, in __init__
    self.smpl_model = SMPL().eval().to(device)
  File "E:\AI\motion-diffusion-model\model\smpl.py", line 72, in __init__
    super(SMPL, self).__init__(**kwargs)
  File "C:\Users\Abdullah\miniconda3\envs\mdm\lib\site-packages\smplx\body_models.py", line 406, in __init__
    **kwargs,
  File "C:\Users\Abdullah\miniconda3\envs\mdm\lib\site-packages\smplx\body_models.py", line 134, in __init__
    smpl_path)
AssertionError: Path ./body_models/smpl\SMPL_NEUTRAL.pkl does not exist!

(mdm) E:\AI\motion-diffusion-model>
DrMemoryFish commented 2 years ago

dont worry, ive fixed the issue, thanks!!!

DrMemoryFish commented 2 years ago

downloaded and extracted this file to "motion-diffusion-model\body_models\"