Improve model card: Add REFLECT paper context and Diffusers tag
#2
by nielsr HF Staff - opened
This PR enhances the model card for the farzadbz/Medical-VAE models by:
- Adding the
library_name: diffusersmetadata tag for improved discoverability and compatibility with thediffusersecosystem, as these VAEs are based on a latent diffusion architecture. - Clarifying their role as foundational components for both the recently published REFLECT: Rectified Flows for Efficient Brain Anomaly Correction Transport paper and the previously cited MAD-AD: Masked Diffusion for Unsupervised Brain Anomaly Detection.
- Including relevant GitHub repository links for both REFLECT and MAD-AD.
- Updating the "Usage" section to include
.sample()on thedecodeoutput, which is common practice for VAEs that return a distribution. - Adding the BibTeX citation for the REFLECT paper to the "Citation & Reference" section.
farzadbz changed pull request status to merged