model
Browse files- scripts/requirements.in +1 -4
scripts/requirements.in
CHANGED
|
@@ -1,6 +1,5 @@
|
|
| 1 |
-
# conda install cuda=12.4 cuda-toolkit=12.4 cuda-compiler=12.4
|
| 2 |
# pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
|
| 3 |
-
# pip install bitsandbytes==0.44.1
|
| 4 |
|
| 5 |
tqdm
|
| 6 |
datasets
|
|
@@ -10,6 +9,4 @@ wandb
|
|
| 10 |
# litgpt[all]
|
| 11 |
litgpt[all] @ git+https://github.com/Lightning-AI/litgpt.git
|
| 12 |
litdata
|
| 13 |
-
# bitsandbytes
|
| 14 |
-
# bitsandbytes==0.44.1
|
| 15 |
grokadamw
|
|
|
|
| 1 |
+
# conda install cuda=12.4 cuda-toolkit=12.4 cuda-compiler=12.4 cuda-cudart=12.4
|
| 2 |
# pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
|
|
|
|
| 3 |
|
| 4 |
tqdm
|
| 5 |
datasets
|
|
|
|
| 9 |
# litgpt[all]
|
| 10 |
litgpt[all] @ git+https://github.com/Lightning-AI/litgpt.git
|
| 11 |
litdata
|
|
|
|
|
|
|
| 12 |
grokadamw
|