TypeError: vqvae_evaluation() got multiple values for argument 'eval_wrapper'

[XzhouCodespace]

I get this error when I am training with train_vqvae. Can you please tell me what is going on.

/home/kemove/code/MotionGPT/train_vqvae.py:156 in │ │ │ │ 153 │ │ 154 │ │ 155 if name == 'main': │ │ ❱ 156 │ main() │ │ 157 │ │ │ │ /home/kemove/code/MotionGPT/train_vqvae.py:117 in main │ │ │ │ 114 │ │ │ avg_recons, avg_perplexity, avg_commit = 0., 0., 0. │ │ 115 │ │ │ 116 │ avg_recons, avg_perplexity, avg_commit = 0., 0., 0. │ │ ❱ 117 │ best_fid, best_div, best_top1, best_top2, best_top3, best_matching, writer, logger = │ │ 118 │ │ │ 119 │ for nb_iter in range(1, args.total_iter + 1): │ │ 120 │ │ gt_motion = next(train_loader_iter) │ │ │ │ /home/kemove/anaconda3/envs/motiongptnew/lib/python3.9/site-packages/torch/utils/_contextlib.py: │ │ 115 in decorate_context │ │ │ │ 112 │ @functools.wraps(func) │ │ 113 │ def decorate_context(*args, **kwargs): │ │ 114 │ │ with ctx_factory(): │ │ ❱ 115 │ │ │ return func(*args, **kwargs) │ │ 116 │ │ │ 117 │ return decorate_context