search

modelscope / scepter

SCEPTER is an open-source framework used for training, fine-tuning, and inference with generative models.
https://github.com/modelscope/scepter
Apache License 2.0
428 stars 22 forks source link

Train Cost #43

Closed mo230761 closed 1 month ago

mo230761 commented 1 month ago

I would like to know the minimum VRAM required to train the StyleBooth model. Could you tell me? Thank you!

hanzhn commented 1 month ago

Total resources cost is very close to SD1.5. Specifically, we use 120 GB VRAM (virtual memory) during training.

mo230761 commented 1 month ago

Thank you very much for your answer. If I set the batch size to 1 and the precision to 16, can I train on a 32G V100?

hanzhn commented 1 month ago

I believe that would working well. Please feel free to re-open the thread if you encounter any other issues.