Skip to content

Refactor HuggingFace Inference Examples#204

Merged
lekurile merged 44 commits intomasterfrom
lekurile/refactor_inference_examples
Oct 13, 2022
Merged

Refactor HuggingFace Inference Examples#204
lekurile merged 44 commits intomasterfrom
lekurile/refactor_inference_examples

Conversation

@lekurile
Copy link
Contributor

@lekurile lekurile commented Oct 12, 2022

This PR reorganizes and refactors the DeepSpeed huggingface inference examples.

Changes in this PR:

  • Remove Transformers folder
  • Add README(s)
    • pip install requirements.txt
    • Code example
    • Point to benchmarking and other resources in DeepSpeed repo
  • Normalize all names i.e. test-[model_name].py
  • Add T5 translation task (English to French)
  • Add huggingface pipeline() object and refactor test-gptj.py
  • Create folders for different types of ML tasks (text-generation, fill-mask, etc)
  • Add BERT fill-mask example
  • Update queries to something more sensible
  • TODO: Add test-bloom to text-generation folder
  • Fix typos in code comments

Copy link
Contributor

@awan-10 awan-10 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks, Lev! This PR looks great to me.

@jeffra and @RezaYazdaniAminabadi -- kindly review it when you get a chance so we can merge it soon.

@arashb arashb requested review from arashb and removed request for arashb October 13, 2022 18:58
@lekurile lekurile merged commit e77b3e4 into master Oct 13, 2022
hwchen2017 pushed a commit that referenced this pull request Jun 8, 2025
This PR reorganizes and refactors the DeepSpeed huggingface inference examples.

Changes in this PR:
- Remove Transformers folder
- Add README(s)
  - pip install requirements.txt
  - Code example
  - Point to benchmarking and other resources in DeepSpeed repo
- Normalize all names i.e. test-[model_name].py
- Add T5 translation task (English to French)
- Add huggingface pipeline() object and refactor test-gptj.py
- Create folders for different types of ML tasks (text-generation, fill-mask, etc)
- Add BERT fill-mask example
- Update queries to something more sensible
- TODO: Add test-bloom to text-generation folder
- Fix typos in code comments

Co-authored-by: Lev Kurilenko <lekurile@microsoft.com>
Co-authored-by: Ammar Ahmad Awan <ammar.awan@microsoft.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants