Improve model card with authors, methodology, and library_name

#1
by nielsr HF Staff - opened

Hi! I'm Niels from the community science team at Hugging Face.

This PR improves the model card for GB-DM by:

  • Adding library_name: pytorch to the metadata. Since the model uses PyTorchModelHubMixin, this allows the Hub to display an automated code snippet for users.
  • Adding the full author list from the paper to the description.
  • Expanding the "Overview" with a "Method Overview" section that summarizes the use of structured variational distributions and dual latent encodings, as described in the paper.
  • Adding a diagram from the official repository to help users visualize the architecture.

These changes help researchers better understand the technical novelty of the grey-box approach and make the model more discoverable on the Hub.

I updated manually with the changes mentioned because of conflicts

GurjeetSinghSangra changed pull request status to closed

Sign up or log in to comment