Add `.to("cuda")` to the example snippet for float16 compatibility 0bbf937 anton-l HF staff commited on Jan 5, 2023