Pin torch version

Victor-Jung · Victor-Jung · commit d5ad963fbde0 · 2026-01-12T15:05:24.000+01:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -18,7 +18,7 @@ authors = [
     {name = "Federico Brancasi", email = "fbrancasi@ethz.ch"},
 ]
 dependencies = [
-    "torch>=2.1.2",
+    "torch==2.1.2",
     "torchvision>=0.16.2",
     "torchaudio>=2.1.2",
     "brevitas==0.11.0",

Original file line number	Diff line number	Diff line change
`@@ -18,7 +18,7 @@ authors = [`
`18`	`18`	`{name = "Federico Brancasi", email = "fbrancasi@ethz.ch"},`
`19`	`19`	`]`
`20`	`20`	`dependencies = [`
`21`		`- "torch>=2.1.2",`
	`21`	`+ "torch==2.1.2",`
`22`	`22`	`"torchvision>=0.16.2",`
`23`	`23`	`"torchaudio>=2.1.2",`
`24`	`24`	`"brevitas==0.11.0",`