.. |
op_binding
|
ce535945e6
fix: change ==NONE to is (#3923)
|
1 年之前 |
triton
|
807d1b5dfc
scripts/check-torchcuda.py: add checking for tensor.is_cuda (#3843)
|
1 年之前 |
__init__.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 年之前 |
bias_add.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 年之前 |
config.py
|
1ba4098918
Fix Stable Diffusion Injection (#4078)
|
1 年之前 |
diffusers_2d_transformer.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 年之前 |
diffusers_attention.py
|
1ba4098918
Fix Stable Diffusion Injection (#4078)
|
1 年之前 |
diffusers_transformer_block.py
|
ce535945e6
fix: change ==NONE to is (#3923)
|
1 年之前 |
ds_attention.py
|
0a61d5d664
Hybrid Engine Refactor and Llama Inference Support (#3425)
|
1 年之前 |
ds_mlp.py
|
15f94ae756
Engine side fix for loading llama checkpoint fine-tuned with zero3 (#3981)
|
1 年之前 |
moe_inference.py
|
ce535945e6
fix: change ==NONE to is (#3923)
|
1 年之前 |
triton_ops.py
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 年之前 |