apple / ml-mdm

Train high-quality text-to-image diffusion models in a data & compute efficient manner
https://machinelearning.apple.com/research/matryoshka-diffusion-models
MIT License
448 stars 31 forks source link

Add doc strings to testcases (25) #28

Closed bdeanhardt closed 1 week ago

bdeanhardt commented 1 week ago

added docstrings to test cases for clarity.

luke-carlson commented 1 week ago

Thank you :+1:

bdeanhardt commented 1 week ago

Hey @bdeanhardt I've approved this PR and moved it out of draft stage.

We can merge it already and if you have additional changes, you could always do it in a follow up PR

Awesome! I think it's fine to merge, I only put it in draft mode incase things needed to be reworded.

luke-carlson commented 1 week ago

The wording looked good to me :+1: