missing DenoisingVIT package

[taiamiti]

Hello @sgraine, thanks for the great work. I want to test the code and i followed the denoised vit installation. However in your code I see that you have a package DenoisedViT imported that have a DenoisingViT.ViTWrapper() function. Can you explain how to install this package ?

Thanks a lot!

[sgraine]

Hi @taiamiti thanks for your interest in our work ! The DenoisingViT package came with the ViTWrapper function as part of it - we didn't install anything else following the normal denoising vit installation. It seems they may have renamed it - perhaps try "PretrainedViTWrapper" instead?