fix env

chromecast56 · chromecast56 · commit 3518b8375d53 · 2024-02-23T12:41:16.000-05:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -13,7 +13,7 @@ classifiers = [
     "License :: OSI Approved :: Apache Software License",
 ]
 dependencies = [
-    "torch", "transformers", "accelerate", "sentencepiece", "protobuf",
+    "torch", "transformers==4.31.0", "accelerate", "sentencepiece", "protobuf",
     "datasets", "triton==2.0.0"
 ]
 

Original file line number	Diff line number	Diff line change
`@@ -13,7 +13,7 @@ classifiers = [`
`13`	`13`	`"License :: OSI Approved :: Apache Software License",`
`14`	`14`	`]`
`15`	`15`	`dependencies = [`
`16`		`- "torch", "transformers", "accelerate", "sentencepiece", "protobuf",`
	`16`	`+ "torch", "transformers==4.31.0", "accelerate", "sentencepiece", "protobuf",`
`17`	`17`	`"datasets", "triton==2.0.0"`
`18`	`18`	`]`
`19`	`19`