shreyask commited on
Commit
5f39b6a
·
verified ·
1 Parent(s): 01ac3f0

Upload pyproject.toml with huggingface_hub

Browse files
Files changed (1) hide show
  1. pyproject.toml +22 -0
pyproject.toml ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [project]
2
+ name = "needle-export"
3
+ version = "0.0.1"
4
+ description = "Export Cactus Needle to ONNX via PyTorch port"
5
+ requires-python = ">=3.11"
6
+ dependencies = [
7
+ "torch",
8
+ "onnx",
9
+ "onnxruntime",
10
+ "jax",
11
+ "jaxlib",
12
+ "flax",
13
+ "sentencepiece",
14
+ "transformers",
15
+ "huggingface_hub",
16
+ "numpy",
17
+ "needle",
18
+ "onnxscript>=0.7.0",
19
+ ]
20
+
21
+ [tool.uv.sources]
22
+ needle = { path = "../external/needle", editable = true }