Query about 1024 Training

[yujiaxu13]

When the subject_1024_beta, did you train it on 1024x1024 images, or simply enlarge the dataset?

[Yuanshi9815]

Hi @yujiaxu13,

The 1024_beta model is available here.

For proper usage, please follow the demo code on Huggingface Space:

1. Use reference images with size (512, 512)
2. Adjust the generate() function code as follows:

   if condition.condition_type == "subject":
       delta = 32 if height == 512 else -32
       condition_ids[:, 2] += delta

Regarding your question about training, yes, we trained the 1024 model on 1024x1024 images specifically. We'll soon release this high-resolution training dataset as a supplement to Subjects200K.