diff options
Diffstat (limited to 'requirements.txt')
| -rw-r--r-- | requirements.txt | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..d7bc501 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,4 @@ +# Runtime tested in the pytorch/pytorch:2.4.1-cuda11.8 image (see README). +torch==2.4.1 +numpy +transformers==4.44.0 # only needed for the GPT-2-scale experiments (gpt2_jlens.py) |