Add support for `torch>=1.8`

[fealho]

PR #30 capped torch to smaller than 1.8. This should be rectified so that we support any version of torch<2, like the rest of the SDV libraries.