Enable using the model on GPU

#3Pull RequestGaaH 创建于 2023-12-21已合并

GaaHcommented

This small change put `x` on the same device as the model parameters.

合并状态:已合并合并于 2023-12-21 关闭于 2023-12-21 0 条评论