model
Browse files- scripts/requirements.in +3 -4
scripts/requirements.in
CHANGED
@@ -1,15 +1,14 @@
|
|
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
|
4 |
|
5 |
tqdm
|
6 |
datasets
|
7 |
jinja2
|
8 |
transformers
|
9 |
-
# bitsandbytes
|
10 |
-
bitsandbytes @ https://github.com/bitsandbytes-foundation/bitsandbytes/releases/download/0.42.0/bitsandbytes-0.42.0-py3-none-any.whl
|
11 |
wandb
|
12 |
# litgpt[all]
|
13 |
litgpt[all] @ git+https://github.com/Lightning-AI/litgpt.git
|
14 |
litdata
|
15 |
-
|
|
|
|
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
|
7 |
jinja2
|
8 |
transformers
|
|
|
|
|
9 |
wandb
|
10 |
# litgpt[all]
|
11 |
litgpt[all] @ git+https://github.com/Lightning-AI/litgpt.git
|
12 |
litdata
|
13 |
+
# bitsandbytes
|
14 |
+
grokadamw
|