Open qixuanwang-233 opened 5 months ago
Stage 3 is pretty useless, it's just a generic upscaler by stability.
does it still oom if you bypass it?
Stage 3 is pretty useless, it's just a generic upscaler by stability.
does it still oom if you bypass it?
Thank you for your response, stage 3 is all I need and I solved this problem by using xFormers.
Good for you! Btw, how do you use xformer with deepfloyd?
*update: found it enable_xformers_memory_efficient_attention() on the pipeline
As in the title, OOM appears when I use DeepFloyd IF through diffuser (I use 32GB video memory).
I used the diffuser example directly:
![image](https://github.com/deep-floyd/IF/assets/71957622/266ace5d-17f5-497a-9790-362de81935eb)