search

showlab / Tune-A-Video

[ICCV 2023] Tune-A-Video: One-Shot Tuning of Image Diffusion Models for Text-to-Video Generation
https://tuneavideo.github.io
Apache License 2.0
4.22k stars 384 forks source link

CUDA error: invalid argument #21

Closed Fitz135 closed 1 year ago

Fitz135 commented 1 year ago

I have met some problems when I run the code. I think maybe there is something wrong with "accelerator.backward(loss)" in line 294 in train_tuneavideo.py. Hope to seek your valuable opinions. Thank you. image

zhangjiewu commented 1 year ago

looks like it is caused by xformers. can you try disabling enable_xformers_memory_efficient_attention in config file?

Fitz135 commented 1 year ago

I installed another version of xfomers. The problem has been solved. Thanks.