Skip to content

Release PegasusFlow on Hugging Face#1

Description

@NielsRogge

Hi @MasterYip 馃

Niels here from the open-source team at Hugging Face. I discovered your work on Arxiv and its corresponding Hugging Face paper page: https://huggingface.co/papers/2509.08435.
The paper page lets people discuss about your paper and lets them find artifacts related to it (your models, datasets, or demo for instance). You can also claim the paper as yours, which will show up on your public profile at HF, and add Github and project page URLs.

I noticed from your GitHub repository (https://github.com/MasterYip/PegasusFlow) and project page (https://masteryip.github.io/pegasusflow.github.io/) that the code and a video for PegasusFlow are "coming soon". That's great news!

Whenever you're ready, it'd be fantastic to make the PegasusFlow model checkpoint available on the 馃 hub, to improve its discoverability/visibility.
We can add tags (like "robotics") so that people find it when filtering https://huggingface.co/models.

Uploading models

See here for a guide: https://huggingface.co/docs/hub/models-uploading.

In this case, we could leverage the PyTorchModelHubMixin class which adds from_pretrained and push_to_hub to any custom nn.Module. Alternatively, one can leverages the hf_hub_download one-liner to download a checkpoint from the hub.

We encourage researchers to push each model checkpoint to a separate model repository, so that things like download stats also work. We can then also link the checkpoints to the paper page.

You can also build a demo for your model on Spaces, we can provide you a ZeroGPU grant, which gives you A100 GPUs for free.

Let me know if you're interested/need any help regarding this, or when you plan to release the code!

Cheers,

Niels
ML Engineer @ HF 馃

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions