demo2 code error

[tengshaofeng]

Thanks for your great job. It is really usefull. But when I run the demo2(), it comes error.
File "/Users/tengbaoqiang/Documents/code/Moving-Least-Squares/demo.py", line 327, in
demo2()
File "/Users/tengbaoqiang/Documents/code/Moving-Least-Squares/demo.py", line 167, in demo2
rigid2 = mls_rigid_deformation_pt(vy, vx, p2, q2, alpha=1)
File "/Users/tengbaoqiang/Documents/code/Moving-Least-Squares/img_utils_pytorch.py", line 250, in mls_rigid_deformation
device = q.device
AttributeError: 'numpy.ndarray' object has no attribute 'device'