zai-org
/

codegeex2-6b

Model card Files Files and versions

compatible with DirectML/ROCm

#5

by davinwang - opened Aug 8, 2023

base: refs/heads/main

←

from: refs/pr/5

Discussion Files changed

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.

compatible with DirectML/ROCm5bc5aff7

Z.ai org Aug 9, 2023

Thanks for fixing this.

Stanislas changed pull request status to merged Aug 9, 2023

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

· Sign up or log in to comment