.. |
__init__.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 year ago |
clip_encoder.py
|
d5a7c1e0b4
Capture short kernel sequences to graph (#4318)
|
10 months ago |
ds_base.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 year ago |
ds_bert.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 year ago |
ds_bloom.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 year ago |
ds_gpt.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 year ago |
ds_llama2.py
|
468882fb68
Add the policy to run llama model from the official repo (#4313)
|
1 year ago |
ds_megatron_gpt.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 year ago |
ds_opt.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 year ago |
ds_transformer.py
|
4a6e0c0624
fix [BUG] 'DeepSpeedGPTInference' object has no attribute 'dtype' for… (#4814)
|
10 months ago |