forked from xiong-zhitong/DOFA-pytorch
-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathpyproject.toml
48 lines (43 loc) · 982 Bytes
/
pyproject.toml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
[build-system]
requires = ["setuptools", "wheel"]
build-backend = "setuptools.build_meta"
[project]
name = "GeoFM"
version = "0.1.0"
description = "Accessible Geospatial Foundation Models with PyTorch Lightning."
authors = [
{ name = "Zhitong Xiong", email = "[email protected]" }
]
maintainers = [
{ name = "Zhitong Xiong", email = "[email protected]" },
{ name = "Nils Lehmann", email = "[email protected]" }
]
dependencies = [
"torch==2.1.2",
"torchvision==0.16.2",
"numpy==1.26.4",
"openmim",
"mmsegmentation==1.2.2",
"mmcv==2.1.0",
"tensorboard",
"loguru",
"timm",
"geobench",
"torchmetrics",
"ftfy",
"regex",
"Cython",
"einops",
"yacs",
"kornia",
"pydantic",
"omegaconf",
"wandb",
"python-dotenv",
"torchgeo @ git+https://github.com/microsoft/torchgeo.git",
"fastparquet",
"mlflow",
"ray>=2.40.0",
"huggingface-hub>=0.27.1",
"peft>=0.14.0"
]