ml-mdm
ml-mdm copied to clipboard
add type hinting to any file that needs it
Bella and I can take this issue.
some type hinting for generate_sample.py: https://github.com/apple/ml-mdm/pull/54
Some more type hinting for generate_sample.py, diffusion.py, and samplers.py https://github.com/apple/ml-mdm/pull/52
added type hinting for NestedSampler, modelEMA, nestedUNET, UNET and Diffusion #62