search

zsyOAOA / ResShift

ResShift: Efficient Diffusion Model for Image Super-resolution by Residual Shifting (NeurIPS@2023 Spotlight, TPAMI@2024)
Other
955 stars 50 forks source link

请问对于fast testing中的测试大概需要多大的显存？ #88

Open Rhhhhhhh opened 3 months ago

Rhhhhhhh commented 3 months ago

我使用的是一张832*480的png图片进行Real-world image super-resolution的测试，使用单卡4090和A6000分别进行测试，参数为scale 4 version v3，一直提示爆显存，大概需要多少显存才够呢？还是说至少需要多少显存？

zsyOAOA commented 3 months ago
  1. 可以调整chop_size(https://github.com/zsyOAOA/ResShift/blob/journal/inference_resshift.py#L53)的大小来调整占用的显存，默认是512.
  2. 检查一下是否安装了xformers, 可以节省很多显存。 @Rhhhhhhh
Rhhhhhhh commented 3 months ago

十分感谢！调整了chop_size之后成功了！ xformers应该是有安装的，是按照readme中的requirements.txt安装的包，应该是有包含的。具体代码细节我再研究研究