Skip to content

Commit

Permalink
Pre-Commit Fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
Helw150 committed Oct 30, 2024
1 parent c9828f9 commit 7067d14
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions config/diva_flash.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ data:
processor: "openai/whisper-large-v3"
configs:
cv:
id: mozilla-foundation/common_voice_17_0
id: mozilla-foundation/common_voice_17_0
cache_dir: "gs://diva-flash/processed/llama/cv"
name: "en"
text_key: "sentence"
Expand Down Expand Up @@ -33,4 +33,4 @@ optimizer:
weight_decay: 0.1
warmup: 0.01
hf_save_path: gs://diva-flash/librispeech-hf-checkpoints
diva_training: true
diva_training: true
2 changes: 1 addition & 1 deletion src/levanter/models/diva.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,8 +30,8 @@


silence_transformer_nag()
from transformers import AutoTokenizer # noqa: E402
from transformers import PretrainedConfig as HfConfig # noqa: E402
from transformers import AutoTokenizer


class DivaHFCheckpointer(HFCheckpointConverter["DivaModel"]):
Expand Down

0 comments on commit 7067d14

Please sign in to comment.