Create requirements.txt
Browse files- requirements.txt +20 -0
requirements.txt
ADDED
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# GPT2-BERT ๋ณํฉ ๋ชจ๋ธ ํ
์คํฐ Requirements
|
2 |
+
|
3 |
+
# ํ์ ๋ผ์ด๋ธ๋ฌ๋ฆฌ
|
4 |
+
torch>=2.0.0
|
5 |
+
transformers>=4.35.0
|
6 |
+
gradio
|
7 |
+
numpy>=1.24.0
|
8 |
+
|
9 |
+
# GPU ์ง์ (์ ํ์ฌํญ)
|
10 |
+
# CUDA 11.8 ๊ธฐ์ค - ๋ณธ์ธ์ CUDA ๋ฒ์ ์ ๋ง๊ฒ ์์
|
11 |
+
# torch==2.0.1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
|
12 |
+
|
13 |
+
# ์ถ๊ฐ ์ ํธ๋ฆฌํฐ
|
14 |
+
huggingface-hub>=0.19.0
|
15 |
+
accelerate>=0.25.0
|
16 |
+
sentencepiece>=0.1.99
|
17 |
+
protobuf>=3.20.0
|
18 |
+
|
19 |
+
# ๋ชจ๋ธ ์ต์ ํ (์ ํ์ฌํญ)
|
20 |
+
bitsandbytes>=0.41.0
|