fix: constrain torch version (#232)

mahone3297 · akaitsuki-ii · gemini-code-assist[bot] · web-flow · commit 74511c363ed8 · 2026-02-12T11:06:44.000+08:00
* fix: constrain torch version and add transformers dependency - Change torch requirement from >= 2.6 to <= 2.8 for version compatibility * Apply suggestion from @gemini-code-assist[bot] Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> --------- Co-authored-by: Akaitsuki <akaitsuki.ii@gmail.com> Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
diff --git a/pyproject.toml b/pyproject.toml
@@ -13,7 +13,7 @@ classifiers = [
 ]
 requires-python = ">=3.10"
 dependencies = [
-    "torch >= 2.6",
+    "torch >= 2.6, <= 2.8",
     "torchvision",
     "numpy",
     "safetensors",

Original file line number	Diff line number	Diff line change
`@@ -13,7 +13,7 @@ classifiers = [`
`13`	`13`	`]`
`14`	`14`	`requires-python = ">=3.10"`
`15`	`15`	`dependencies = [`
`16`		`- "torch >= 2.6",`
	`16`	`+ "torch >= 2.6, <= 2.8",`
`17`	`17`	`"torchvision",`
`18`	`18`	`"numpy",`
`19`	`19`	`"safetensors",`