MinishLab
diff --git a/‎pyproject.toml‎
Lines changed: 1 addition & 1 deletion b/‎pyproject.toml‎
Lines changed: 1 addition & 1 deletion
@@ -60,7 +60,7 @@ dev = [
     "ruff",
 ]
 
-distill = ["torch", "transformers", "scikit-learn"]
+distill = ["torch<2.8.0", "transformers", "scikit-learn"]
 onnx = ["onnx", "torch"]
 # train also installs inference
 train = ["torch", "lightning", "scikit-learn", "skops"]
Original file line number	Diff line number	Diff line change
`@@ -60,7 +60,7 @@ dev = [`
`60`	`60`	`"ruff",`
`61`	`61`	`]`
`62`	`62`
`63`		`-distill = ["torch", "transformers", "scikit-learn"]`
	`63`	`+distill = ["torch<2.8.0", "transformers", "scikit-learn"]`
`64`	`64`	`onnx = ["onnx", "torch"]`
`65`	`65`	`# train also installs inference`
`66`	`66`	`train = ["torch", "lightning", "scikit-learn", "skops"]`