mazesmazes commited on
Commit
a0d385f
·
verified ·
1 Parent(s): 5b70822

Update custom model files, README, and requirements

Browse files
Files changed (1) hide show
  1. asr_config.py +3 -3
asr_config.py CHANGED
@@ -45,12 +45,12 @@ class ASRConfig(transformers.PretrainedConfig):
45
  # Set default generation parameters
46
  generation_defaults = {
47
  "num_beams": 1,
48
- "max_new_tokens": 96,
49
  "min_new_tokens": 1,
50
  "do_sample": False,
51
- "repetition_penalty": 1.05,
52
  "length_penalty": 1.0,
53
- "no_repeat_ngram_size": 3,
54
  "use_cache": True,
55
  }
56
 
 
45
  # Set default generation parameters
46
  generation_defaults = {
47
  "num_beams": 1,
48
+ "max_new_tokens": 64,
49
  "min_new_tokens": 1,
50
  "do_sample": False,
51
+ "repetition_penalty": 1.0,
52
  "length_penalty": 1.0,
53
+ "no_repeat_ngram_size": 0,
54
  "use_cache": True,
55
  }
56