x54-729's picture
Add torch_dtype to README example
509b748