зеркало из https://github.com/microsoft/DeepSpeed.git
1d8196736f
Fixing following error
/datadisk2/wengshiy/llm.devkit/DeepSpeed/deepspeed/runtime/utils.py
return get_accelerator().FloatTensor(float(v)).detach()
TypeError: new(): data must be a sequence (got float)
cuda accelerator modified the interface for fixing warning:
|
||
|---|---|---|
| .. | ||
| __init__.py | ||
| abstract_accelerator.py | ||
| cpu_accelerator.py | ||
| cuda_accelerator.py | ||
| hpu_accelerator.py | ||
| mps_accelerator.py | ||
| npu_accelerator.py | ||
| real_accelerator.py | ||
| xpu_accelerator.py | ||