Skip to content

Latest commit

 

History

History
23 lines (18 loc) · 407 Bytes

File metadata and controls

23 lines (18 loc) · 407 Bytes

DeepSpeed Huggingface Translation Examples

Setup

Python dependencies:

pip install -r requirements.txt

Usage

Examples can be run as follows:

deepspeed --num_gpus [number of GPUs] test-[model].py

Example Output

Command:

deepspeed --num_gpus 1 test-t5-base.py

Output:

[{'translation_text': 'Le renard brun rapide saute au-dessus du chien lazy.'}]