Skip to content

Conversation

@guangy10
Copy link
Contributor

@guangy10 guangy10 commented Oct 26, 2024

With more models enabled on HuggingFace, extend the script to export and lower different tasks of models from HuggingFace.

  • For CausalLM
    python -m extension.export_util.export_hf_model -hfm="Qwen/Qwen2.5-0.5B" --task=causal_lm

  • For MaskedLM
    python -m extension.export_util.export_hf_model -hfm="FacebookAI/roberta-base" --task="masked_lm"

  • For SemanticSegmentation
    python -m extension.export_util.export_hf_model -hfm="Intel/dpt-large-ade" --task="semantic_segmentation"

  • For DepthEstimation
    python -m extension.export_util.export_hf_model -hfm="LiheYoung/depth-anything-small-hf" --task="depth_estimation"

@guangy10 guangy10 requested a review from kirklandsign October 26, 2024 02:10
@pytorch-bot
Copy link

pytorch-bot bot commented Oct 26, 2024

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/6509

Note: Links to docs will display an error until the docs builds have been completed.

✅ No Failures

As of commit 25c150c with merge base e93ad5f (image):
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Oct 26, 2024
@guangy10 guangy10 force-pushed the hf_export_masked_lm branch from 1b48e83 to 6842610 Compare October 26, 2024 02:12
@facebook-github-bot
Copy link
Contributor

@guangy10 has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

@facebook-github-bot
Copy link
Contributor

@guangy10 has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

@guangy10 guangy10 force-pushed the hf_export_masked_lm branch from e27ccdc to c8e8b75 Compare October 31, 2024 23:41
@guangy10 guangy10 force-pushed the hf_export_masked_lm branch from c8e8b75 to 25c150c Compare October 31, 2024 23:53
@guangy10 guangy10 closed this Jun 6, 2025
@guangy10 guangy10 deleted the hf_export_masked_lm branch June 6, 2025 22:06
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants