[Question] VRAM for distillation

segmind / SSD-1B

SSD-1B, an open-source text-to-image model, outperforming previous versions by being 50% smaller and 60% faster than SDXL.

Apache License 2.0

164 stars 14 forks source link

Open alfredplpl opened 10 months ago

alfredplpl commented 10 months ago

I'm trying do distillation of the my sdxl model with A6000 (48GB). But, I cannot do distillation with 1024x1024 because of OOM.

How much VRAM do we need for distillation?

Thanks in advance.

huan085128 commented 10 months ago

A100 80G is not working too.