pschaldenbrand / anipainter Goto Github PK

Hi,
First at all, this is absolutely a piece of great work! I love the results of the AniPainter and I was trying to implement the paint.py file in Colab.
This is the issue I got when implementing it. I am new in python, can you please help me on this?

Traceback (most recent call last):
File "paint.py", line 655, in
output_dir=args.output_dir, w=args.w, h=args.h)
File "paint.py", line 594, in paint_fn
save_strokes_for_robot(s, img, painting.cpu().numpy(), allowed_colors, ignore_whites=ignore_whites, output_dir=output_dir)
File "paint.py", line 182, in save_strokes_for_robot
S_whites = np.array(copy.deepcopy(S))
File "/usr/local/lib/python3.7/dist-packages/torch/_tensor.py", line 732, in array
return self.numpy()
TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.