compatible with DirectML/ROCm
#5
by
davinwang
- opened
Tensor.new is a deprecated constructor and does not support PrivateUse1 in pytorch 1.13.1/2.0.0, use torch.ones() instead. Please refer to https://github.com/microsoft/DirectML/issues/400 and https://github.com/pytorch/pytorch/issues/95734 and https://huggingface.co/THUDM/chatglm2-6b/discussions/71 for more detail. This should also fix the ROCm compatibility in this file.
Thanks for fixing this.
Stanislas
changed pull request status to
merged