diff options
Diffstat (limited to 'config.yaml')
| -rw-r--r-- | config.yaml | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/config.yaml b/config.yaml index e5ff43a..41b473c 100644 --- a/config.yaml +++ b/config.yaml @@ -31,4 +31,7 @@ checkpoints: inference: model_load_path: "YOUR/PATH" # path to load model from beam_width: 10 # beam width for beam search - device: "cuda" # device to run inference on if gpu is available, else "cpu" will be set automatically
\ No newline at end of file + device: "cuda" # device to run inference on if gpu is available, else "cpu" will be set automatically + +lang_model: + path: "data/mls_lm_german" #path where model and supplementary files are stored |