NVIDIA-NeMo
diff --git a/‎pyproject.toml‎
Lines changed: 1 addition & 0 deletions b/‎pyproject.toml‎
Lines changed: 1 addition & 0 deletions
@@ -77,6 +77,7 @@ dependencies = [
     "einops",
     "ijson",
     "pyarrow>=20.0.0,<21.0.0",
+    "torchao>=0.14.1"
 ]
 
 [project.optional-dependencies]
Original file line number	Diff line number	Diff line change
`@@ -77,6 +77,7 @@ dependencies = [`
`77`	`77`	`"einops",`
`78`	`78`	`"ijson",`
`79`	`79`	`"pyarrow>=20.0.0,<21.0.0",`
	`80`	`+ "torchao>=0.14.1"`
`80`	`81`	`]`
`81`	`82`
`82`	`83`	`[project.optional-dependencies]`