bugfix: fix cuda: + pytorch latest version, input type and weight type should be the same b4c2583 soulteary commited on May 23, 2022