Skip to content

Commit f3dec5b

Browse files
committed
Cap dependency versions (litert 1.2)
1 parent 150232d commit f3dec5b

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

setup.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -85,14 +85,14 @@
8585
"scipy",
8686
"safetensors",
8787
"tabulate",
88-
"torch>=2.4.0",
89-
"tf-nightly>=2.19.0.dev20250101",
90-
"ai-edge-litert-nightly",
91-
"ai-edge-quantizer-nightly",
88+
"torch>=2.4.0,<2.7.0",
89+
"tensorflow==2.19.*",
90+
"ai-edge-litert==1.2.*",
91+
"ai-edge-quantizer==0.1.*",
9292
"jax",
9393
"torch-xla2[odml]>=0.0.1.dev20241201",
9494
],
9595
extras_require={
96-
"torch-xla": ["torch_xla>=2.4.0"],
96+
"torch-xla": ["torch_xla>=2.4.0,<2.7.0"],
9797
},
9898
)

0 commit comments

Comments
 (0)