From 271c81b7344d24edb36155eae475db19c1ed0980 Mon Sep 17 00:00:00 2001 From: Seo Chaegeon Date: Tue, 9 Dec 2025 17:12:09 +0900 Subject: [PATCH] Update PyTorch dependencies in requirements.txt Removed comments about CUDA version and updated PyTorch dependencies. --- requirements.txt | 15 +++++++++++++++ 1 file changed, 15 insertions(+) create mode 100644 requirements.txt diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..0de291e --- /dev/null +++ b/requirements.txt @@ -0,0 +1,15 @@ +fastapi +uvicorn[standard] +pydantic +python-multipart + +transformers==4.37.2 +tokenizers +sentencepiece + +torch==2.2.1+cu121 +torchvision==0.17.1+cu121 +torchaudio==2.2.1+cu121 +--extra-index-url https://download.pytorch.org/whl/cu121 + +numpy