search

bghira / SimpleTuner

A general fine-tuning kit geared toward diffusion models.
GNU Affero General Public License v3.0
1.49k stars 131 forks source link

How to train 900M PixArt Sigma ? #580

Closed kingkong135 closed 1 month ago

kingkong135 commented 1 month ago

I have a question, I see the Pixart Sigma train by default is 600M, I want to try finetune with 900M PixArt Sigma like PixArt Sigma 900M: Enhanced Text-to-Image Model

bghira commented 1 month ago

you'll need to have a version from hugging face hub. and then just use that as the MODEL_NAME, eg. terminusresearch/pixart-900m-1024-ft-v0.6