Multi Class Model Training - SageMaker Jumpstart Issue

Hey there, looks like you are trying to train your stable diffusion model on a multi class training set. Unfortunately this is not a feature in SageMaker JumpStart just yet. There is a way to get around this however which follows the strategy of

1. Fine-tune the model for Subject A.
2. Fine-tune the resulting model from Step 1 for Subject B.
3. Generate images of Subject A and Subject B using the output model from Step 2.

If you’re after more information / an example, check out this blog and scroll down to fine tuning considerations. It goes through the process of training a stable diffusion model with one class then runs through how it has trained it on multiple classes and the associated strengths and weakness of both approaches. Hope this helps!