get_time_embed() takes 1 positional argument but 2 were given

[jinganglang567]

sorry to bother you ,when i train the code
Traceback (most recent call last):
File "train.py", line 283, in
loss = get_loss(data) / 4
File "train.py", line 271, in get_loss
out_unet = unet(out_vae=out_vae_noise,
File "/root/miniconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "/root/autodl-tmp/Diffusion_From_Scratch-main/Unet.py", line 385, in forward
time = self.get_time_embed(time)
TypeError: get_time_embed() takes 1 positional argument but 2 were given