How much VRAM required for Share Captioner?

PixArt-alpha / PixArt-sigma

PixArt-Σ: Weak-to-Strong Training of Diffusion Transformer for 4K Text-to-Image Generation

https://pixart-alpha.github.io/PixArt-sigma-project/

GNU Affero General Public License v3.0

1.44k stars 67 forks source link

How much VRAM required for Share Captioner? #103

Closed Han230104 closed 4 weeks ago

Han230104 commented 1 month ago

Hi author, great work! I saw you use the Share-Captioner to label the image. I wonder how much memory does it take to run the inference ? Looking forward to your reply !

Han230104 commented 4 weeks ago

I am able to run the share-captioner with A6000 (48G). 4090 (24G) is not enough.