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
|