[Dreambooth] Editable number of class images

[Mr-Philo]

Recently the code in class DreamboothDataset will directly make num of class_images to the length of the class_images folder:

self.num_class_images = len(self.class_images_path)

However, if user want to test different effects with different class images (for example user want to test 50 but the folder contains 100), it will ignore the number 50 and still use 100, thus fail to handle it. So, it seems better to modify the code to:

self.num_class_images = min(len(self.class_images_path), class_num)

All three dreambooth training scripts in this PR have been modified in the same way.

[HuggingFaceDocBuilderDev]

The documentation is not available anymore as the PR was closed or merged.

[github-actions]

This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread.

Please note that issues that do not follow the contributing guidelines are likely to be ignored.

[patrickvonplaten]

From a quick glance this makes sense to me - @patil-suraj could you quickly check this?

[patil-suraj]

Sorry to be so late here, makes sense, looks good!

[patil-suraj]

This should not be None actually, otherwise the min call will fail. When it's None we should set it to the value of len(self.class_images_path)

[Mr-Philo]

Yeah, that's my mistake. Change the code like this:

if class_num is not None:
    self.num_class_images = min(len(self.class_images_path), class_num)
else:
    self.num_class_images = len(self.class_images_path)

[Mr-Philo]

Hi all, thanks for checking this PR and reviewing the code! I have committed the new code😊

[patrickvonplaten]

Cool, good for merge then!