search

pixeli99 / TrackDiffusion

Official PyTorch implementation of TrackDiffusion (https://arxiv.org/abs/2312.00651)
61 stars 4 forks source link

SVD分支的代码是完整的吗 #9

Closed donghaotian123 closed 2 weeks ago

donghaotian123 commented 2 weeks ago

我尝试运行SVD分支的文件，但此分支缺少third party文件夹，main分支的diffusers中似乎缺少unet代码，当我下载最新版本0.30.0的diffusers时，forward函数又缺少cross_attention_kwargs这个参数

pixeli99 commented 2 weeks ago

我这边是diffusers 0.24.0的分支，SVD分支确实缺少了unet部分的代码修改，你可以从google drive上下载这部分的修改代码，我最近会更新SVD分支的代码

https://drive.google.com/drive/folders/1pk4lYYMq_oAvBN4BkdILXey4iPKhKOX1?usp=sharing

donghaotian123 commented 2 weeks ago

好的，我的代码已经能正常训练测试了，谢谢您 : )