bmay commited on
Commit
9181593
1 Parent(s): e35c029

Add pyproject.toml

Browse files
Files changed (1) hide show
  1. pyproject.toml +30 -0
pyproject.toml ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [build-system]
2
+ requires = ["setuptools>=61.0"]
3
+ build-backend = "setuptools.build_meta"
4
+
5
+ [project]
6
+ name = "theia"
7
+ version = "0.1.0"
8
+ dependencies = [
9
+ "numpy>=1.26.4",
10
+ "torch>=2.1.0",
11
+ "torchaudio>=0.13.1",
12
+ "torchvision>=0.16.0",
13
+ "tensorflow==2.15.1",
14
+ "hydra-core>=1.3.2",
15
+ "omegaconf>=2.3.0",
16
+ "rich>=13.4.2",
17
+ "wandb",
18
+ "transformers",
19
+ "einops",
20
+ "webdataset@git+https://github.com/elicassion/webdataset.git@elicassion/fix_shuffle_bug",
21
+ "opencv-python==4.8.0.74",
22
+ "av",
23
+ "seaborn",
24
+ "matplotlib",
25
+ "datasets",
26
+ "tqdm",
27
+ ]
28
+
29
+ [tool.setuptools.packages.find]
30
+ where = ["src"]