Skip to content

Add BLOOM huggingface inference example#211

Merged
lekurile merged 12 commits intomasterfrom
lekurile/bloom-3b-example
Oct 28, 2022
Merged

Add BLOOM huggingface inference example#211
lekurile merged 12 commits intomasterfrom
lekurile/bloom-3b-example

Conversation

@awan-10
Copy link
Contributor

@awan-10 awan-10 commented Oct 27, 2022

No description provided.

@lekurile
Copy link
Contributor

This PR adds a bloom inference example (bigscience/bloom-3b) and a corresponding helper Pipeline class meant to mimic the functionality and API of the huggingface pipelines. This class was added in order to comprehend bloom meta tensors and checkpoint loading in a more organized way, that closely matched the existing examples.

This PR also cleans up extra whitespace across the inference examples.

@lekurile lekurile changed the title Lekurile/bloom 3b example Add BLOOM huggingface inference example Oct 28, 2022
@lekurile lekurile self-requested a review October 28, 2022 17:59
@lekurile lekurile merged commit 7d0260b into master Oct 28, 2022
hwchen2017 pushed a commit that referenced this pull request Jun 8, 2025
This PR adds a bloom inference example (bigscience/bloom-3b) and a corresponding helper Pipeline class meant to mimic the functionality and API of the huggingface pipelines. This class was added in order to comprehend bloom meta tensors and checkpoint loading in a more organized way, that closely matched the existing examples.

This PR also cleans up extra whitespace across the inference examples.

Co-authored-by: Lev Kurilenko <lekurile@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