Skip to content

Commit fdda330

Browse files
authored
Fix deprecations and update dependencies for examples/albert (#595)
* Install setuptools+wheel in develop mode during CI * Fix deprecations and update dependencies for examples/albert
1 parent 2873252 commit fdda330

File tree

4 files changed

+6
-6
lines changed

4 files changed

+6
-6
lines changed

.github/workflows/run-tests.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ jobs:
6060
key: Key-v1-3.8-${{ hashFiles('requirements.txt') }}-${{ hashFiles('requirements-dev.txt') }}
6161
- name: Install dependencies
6262
run: |
63-
python -m pip install --upgrade pip
63+
python -m pip install --upgrade pip setuptools wheel
6464
pip install -r requirements.txt
6565
pip install -r requirements-dev.txt
6666
- name: Build hivemind
@@ -88,7 +88,7 @@ jobs:
8888
key: Key-v1-3.8-${{ hashFiles('requirements.txt') }}-${{ hashFiles('requirements-dev.txt') }}
8989
- name: Install dependencies
9090
run: |
91-
python -m pip install --upgrade pip
91+
python -m pip install --upgrade pip setuptools wheel
9292
pip install -r requirements.txt
9393
pip install -r requirements-dev.txt
9494
- name: Build bitsandbytes

examples/albert/arguments.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,7 @@ class DatasetArguments:
113113
)
114114
tokenizer_path: Optional[str] = field(default="data/tokenizer", metadata={"help": "Path to the tokenizer"})
115115
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",
117117
metadata={"help": "Path to the model config"},
118118
)
119119
cache_dir: Optional[str] = field(default="data", metadata={"help": "Path to the cache"})

examples/albert/requirements.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
1-
transformers==4.6.0
2-
datasets==1.5.0
1+
transformers~=4.6
2+
datasets~=1.5
33
torch_optimizer==0.1.0
44
wandb==0.10.26
55
sentencepiece

examples/albert/run_training_monitor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ class TrainingMonitorArguments(BaseTrainingArguments):
4646
default=5, metadata={"help": "Frequency (in steps) of fetching and saving state from peers"}
4747
)
4848
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",
5050
metadata={"help": "Path to the model config"},
5151
)
5252
repo_path: Optional[str] = field(

0 commit comments

Comments
 (0)