diff --git a/Cargo.toml b/Cargo.toml index 3103ddc..f7554c7 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -15,7 +15,7 @@ default = ["tch"] download-libtorch = ["tch/download-libtorch"] [dependencies] -tch = { version = "0.20", optional = true } +tch = { version = "0.22", optional = true } einops-macros = { path = "einops-macros", version = "0.1.0-alpha.2" } [package.metadata.docs.rs]