From ce7b5723c03268284bde4f9982d5adc0b72c073f Mon Sep 17 00:00:00 2001
From: dh <46441813+DeanHnter@users.noreply.github.com>
Date: Sat, 3 May 2025 22:24:00 +0200
Subject: [PATCH] Create requirements.txt

Added a basic requirements file. Its somewhat difficult now to get this project running as installing the latest dependencies leads to package conflicts when pulling the latest version from pypi. I propose adding this as a minimal set of versioned dependencies to make it simpler for those looking to fine-tune CodeT5+ following the guide.
---
 CodeT5+/requirements.txt | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)
 create mode 100644 CodeT5+/requirements.txt

diff --git a/CodeT5+/requirements.txt b/CodeT5+/requirements.txt
new file mode 100644
index 0000000..7ce440f
--- /dev/null
+++ b/CodeT5+/requirements.txt
@@ -0,0 +1,22 @@
+# ----------  PyTorch CUDA‑12.1 wheels  -----------
+--extra-index-url https://download.pytorch.org/whl/cu121
+torch==2.2.2+cu121
+torchvision==0.17.2+cu121
+torchaudio==2.2.2+cu121
+              
+# ----------  Core utilities  ---------------------
+numpy==1.24.4
+tqdm==4.67.1
+numpy==1.24.4
+
+# ----------  Hugging‑Face ecosystem --------------
+transformers==4.36.2
+accelerate==0.21.0
+datasets==3.5.1
+
+# ----------  DeepSpeed  --------------------------
+deepspeed==0.16.7
+
+# ----------  Logging / evaluation  --------------
+tensorboardX==2.6.2.2
+human_eval==1.0.3