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