model
Browse files- scripts/model.yaml +1 -1
- scripts/requirements.in +4 -2
scripts/model.yaml
CHANGED
|
@@ -119,7 +119,7 @@ optimizer:
|
|
| 119 |
lr: 1.0e-4
|
| 120 |
|
| 121 |
# (type: float, default: 0.01)
|
| 122 |
-
weight_decay: 0.
|
| 123 |
|
| 124 |
# (type: tuple, default: (0.9,0.999))
|
| 125 |
betas:
|
|
|
|
| 119 |
lr: 1.0e-4
|
| 120 |
|
| 121 |
# (type: float, default: 0.01)
|
| 122 |
+
weight_decay: 0.01
|
| 123 |
|
| 124 |
# (type: tuple, default: (0.9,0.999))
|
| 125 |
betas:
|
scripts/requirements.in
CHANGED
|
@@ -1,5 +1,7 @@
|
|
| 1 |
-
#
|
| 2 |
-
#
|
|
|
|
|
|
|
| 3 |
tqdm
|
| 4 |
datasets
|
| 5 |
jinja2
|
|
|
|
| 1 |
+
# conda install numpy==1.26.4 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 --force-reinstall 'https://github.com/bitsandbytes-foundation/bitsandbytes/releases/download/continuous-release_main/bitsandbytes-0.44.2.dev0-py3-none-manylinux_2_24_x86_64.whl'
|
| 4 |
+
|
| 5 |
tqdm
|
| 6 |
datasets
|
| 7 |
jinja2
|