Skip to content

Commit

Permalink
Add notebook of using DeepSpeed with Huggingface
Browse files Browse the repository at this point in the history
  • Loading branch information
YeonwooSung committed Dec 14, 2022
1 parent 27d875d commit 9c0c427
Show file tree
Hide file tree
Showing 2 changed files with 1,481 additions and 1 deletion.
4 changes: 3 additions & 1 deletion DistributedTraining/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,9 @@ It implements everything that are described in [ZeRO paper [1]](https://arxiv.or

### Integrate Huggingface with Deepspeed

[This page](https://huggingface.co/docs/transformers/main_classes/deepspeed?highlight=deepspeed#deepspeed-integration) contains the descriptions and codes for integrating the Deepspeed with Huggingface for distributed training with Huggingface models.
[This page](https://huggingface.co/docs/transformers/main_classes/deepspeed) contains the descriptions and codes for integrating the Deepspeed with Huggingface for distributed training with Huggingface models.

[This](./src/Huggingface_DeepSpeed_CLI.ipynb) is the example notebook of using Deepspeed with Huggingface.

### DeepSpeed ZeRO

Expand Down
Loading

0 comments on commit 9c0c427

Please sign in to comment.