Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Generalize meta tensor pipeline #199

Merged
merged 10 commits into from
Jun 26, 2023
Merged

Conversation

mrwyattii
Copy link
Contributor

@mrwyattii mrwyattii commented Jun 1, 2023

DeepSpeed-Inference has supported meta tensor loading with BLOOM, OPT, GPT-J, and GPT-NEOX models for some time, but this functionality was never extended to MII. Adding that support here by refactoring the BloomPipeline to work with other models.

Motiviated by #198

Example of added meta_tensor option in mii_config:

import mii
mii_configs = {"tensor_parallel": 8, "dtype": "fp16", "meta_tensor": True}
mii.deploy(task="text-generation", model="facebook/opt-66b", deployment_name="opt", mii_config=mii_configs)

@tohtana tohtana self-requested a review June 26, 2023 17:24
@mrwyattii mrwyattii merged commit 60a85dc into main Jun 26, 2023
4 checks passed
@mrwyattii mrwyattii deleted the mrwyattii/extend-llm-provider branch July 17, 2023 21:34
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.

None yet

2 participants