File tree 4 files changed +6
-6
lines changed
4 files changed +6
-6
lines changed Original file line number Diff line number Diff line change 60
60
key : Key-v1-3.8-${{ hashFiles('requirements.txt') }}-${{ hashFiles('requirements-dev.txt') }}
61
61
- name : Install dependencies
62
62
run : |
63
- python -m pip install --upgrade pip
63
+ python -m pip install --upgrade pip setuptools wheel
64
64
pip install -r requirements.txt
65
65
pip install -r requirements-dev.txt
66
66
- name : Build hivemind
88
88
key : Key-v1-3.8-${{ hashFiles('requirements.txt') }}-${{ hashFiles('requirements-dev.txt') }}
89
89
- name : Install dependencies
90
90
run : |
91
- python -m pip install --upgrade pip
91
+ python -m pip install --upgrade pip setuptools wheel
92
92
pip install -r requirements.txt
93
93
pip install -r requirements-dev.txt
94
94
- name : Build bitsandbytes
Original file line number Diff line number Diff line change @@ -113,7 +113,7 @@ class DatasetArguments:
113
113
)
114
114
tokenizer_path : Optional [str ] = field (default = "data/tokenizer" , metadata = {"help" : "Path to the tokenizer" })
115
115
config_path : Optional [str ] = field (
116
- default = "https://s3.amazonaws.com/models.huggingface.co/bert/ albert-large-v2-config.json " ,
116
+ default = "albert-large-v2" ,
117
117
metadata = {"help" : "Path to the model config" },
118
118
)
119
119
cache_dir : Optional [str ] = field (default = "data" , metadata = {"help" : "Path to the cache" })
Original file line number Diff line number Diff line change 1
- transformers == 4.6.0
2
- datasets == 1.5.0
1
+ transformers ~= 4.6
2
+ datasets ~= 1.5
3
3
torch_optimizer == 0.1.0
4
4
wandb == 0.10.26
5
5
sentencepiece
Original file line number Diff line number Diff line change @@ -46,7 +46,7 @@ class TrainingMonitorArguments(BaseTrainingArguments):
46
46
default = 5 , metadata = {"help" : "Frequency (in steps) of fetching and saving state from peers" }
47
47
)
48
48
model_config_path : str = field (
49
- default = "https://s3.amazonaws.com/models.huggingface.co/bert/ albert-large-v2-config.json " ,
49
+ default = "albert-large-v2" ,
50
50
metadata = {"help" : "Path to the model config" },
51
51
)
52
52
repo_path : Optional [str ] = field (
You can’t perform that action at this time.
0 commit comments