commit fbbdcf8774f7768b1db2927abc0f3140531f6f62 parent e766fc2acc91f39c0227f806eb67c03815653914 Author: Sondre Wold <sondrewold1@gmail.com> Date: Mon, 3 Jun 2024 14:23:06 +0200 Add requirements.txt Diffstat:
| A | requirements.txt | | | 12 | ++++++++++++ |
1 file changed, 12 insertions(+), 0 deletions(-)
diff --git a/requirements.txt b/requirements.txt @@ -0,0 +1,12 @@ +jsonlines==4.0.0 +networkx==3.2.1 +numpy==1.26.3 +safetensors==0.4.2 +tokenizers==0.15.1 +torch==1.11.0+rocm4.5.2 +torch-cluster==1.6.3 +torch-scatter==2.1.2 +torch-sparse==0.6.18 +torch-spline-conv==1.2.2 +torch_geometric==2.4.0 +transformers==4.37.1