deepspeedai / deepspeedai/DeepSpeedExamples
is there any expample for zero-inference
Open
@tjruwase is already working on this.
Since Apr 28, 2023.
- Dominant language
- Python
- Stars
- 6.8k
- Forks
- 1.1k
- Avg merge
- 2d 16h
- Merged PRs (30d)
- 1
Description
when i use zero-inference in deepspeed.init_inference , it is not work
this is my code
deepspeed_config = {
"zero": {
"stage": 3,
"offload_optimizer": {
"device": "cpu"
},
"offload_param": {
"device": "cpu"
},
}
}
self.model = deepspeed.init_inference(
self.model,
config = deepspeed_config
mp_size=get_world_size(),
base_dir=downloaded_model_path,
dtype=args.dtype,
checkpoint=checkpoints_json,
replace_with_kernel_inject=True,
)
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.