Triton Pytorch Get Model