Hi
I've just started playing around with Dance Diffusion, and I'm having the time of my life!
I've trained a model and had no problems with the Fine Tuning Notebook. However, I've encountered an error with 'arg' on the main Dance Diffusion Notebook.
Please see bottom panel for the final error.
Please forgive my ignorance...I suspect it's a simple problem, but I'm not a Python coder. Thank you in advance for your help.

