x54-729's picture
Add torch_dtype to README example
2b91a35