File size: 1,179 Bytes
41dd7e0 | 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 | [project]
name = "coarse-grained-forecasting"
version = "1.0.0"
description = "Direct SI, Cascaded SI and EDM for conditional forecasting from coarse-grained observations: the code of the paper's models, training, sampling and csFID evaluation."
readme = "README.md"
license = { file = "LICENSE" }
requires-python = ">=3.10,<3.11"
dependencies = [
"torch==2.1.2",
"torchvision==0.16.2",
"numpy==1.26.2",
"scipy==1.15.3",
"pyyaml==6.0.1",
"pandas==2.2.1",
"matplotlib==3.8.2",
"scikit-learn==1.3.2",
"h5py==3.15.1",
"einops==0.8.1",
"Pillow==10.0.1",
"clean-fid==0.1.35",
"requests",
]
[tool.uv]
# PyTorch 2.1.2 built for CUDA 11.8 (the environment of the recorded Euler, NLKG and passive-scalar values);
# on a machine without CUDA the same wheels run on the CPU.
[[tool.uv.index]]
name = "pytorch-cu118"
url = "https://download.pytorch.org/whl/cu118"
explicit = true
[tool.uv.sources]
torch = { index = "pytorch-cu118" }
torchvision = { index = "pytorch-cu118" }
[build-system]
requires = ["setuptools>=61"]
build-backend = "setuptools.build_meta"
[tool.setuptools.packages.find]
include = ["bifurcation*", "reproduce*"]
|