Skip to content

Commit 5f9a930

Browse files
committed
change build-system to setuptools instead of poetry; support HF_HOME EnvVar if set.
1 parent 7543b3f commit 5f9a930

File tree

6 files changed

+73
-716
lines changed

6 files changed

+73
-716
lines changed

README.md

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -55,19 +55,20 @@ Visit [TMVisDB](https://tmvisdb.predictprotein.org) [3] to see precomputed predi
5555
If you want to use GPU acceleration (highly recommended), please install the corresponding version of PyTorch: [Get Started](https://pytorch.org/get-started/locally/)
5656

5757
_Reproducibility:_ TMbed tries to make predictions on GPU as deterministic as possible.\
58-
However, some things are left to the user, such as setting the [CUBLAS_WORKSPACE_CONFIG](https://docs.nvidia.com/cuda/cublas/index.html#cublasApi_reproducibility) EnvVar.\
58+
However, some things are left to the user, such as setting the [CUBLAS_WORKSPACE_CONFIG](https://docs.nvidia.com/cuda/cublas/index.html#results-reproducibility) EnvVar.\
5959
Regarding reproducibility in PyTorch, we also recommend their [notes on randomness and reproducibility](https://pytorch.org/docs/stable/notes/randomness.html#reproducibility).
6060

6161
### Requirements
6262

6363
```txt
64-
python = "^3.11"
65-
h5py = "^3.13"
66-
numpy = "^2.2"
67-
torch = "^2.5"
68-
tqdm = "^4.67"
69-
transformers = "^4.50"
70-
typer = "^0.15"
64+
python >= "3.9"
65+
h5py >= "3.2.1"
66+
numpy >= "1.20.3"
67+
sentencepiece >= "0.1.96"
68+
torch >= "1.10.2"
69+
tqdm >= "4.62.3"
70+
transformers >= "4.11.3"
71+
typer >= "0.4.1"
7172
```
7273

7374
## Usage

0 commit comments

Comments
 (0)