some problem with cpu

#3
by zhangbo2008 - opened

if use cpu, no half functon raised error.
好像是缺少half方法.

Change line 90 of file "modeling_chatglm.py" to the following:
inv_freq = 1.0 / (10000 ** (torch.arange(0, dim, 2, device=device) / dim))

Knowledge Engineering Group (KEG) & Data Mining at Tsinghua University org

Fixed.

zxdu20 changed discussion status to closed

Sign up or log in to comment