jhj0517 commited on
Commit
fdb3baa
·
1 Parent(s): b6032b5

Handle none values

Browse files
modules/whisper/whisper_parameter.py CHANGED
@@ -281,7 +281,7 @@ class WhisperValues:
281
  data = {
282
  "whisper": {
283
  "model_size": self.model_size,
284
- "lang": self.lang,
285
  "is_translate": self.is_translate,
286
  "beam_size": self.beam_size,
287
  "log_prob_threshold": self.log_prob_threshold,
@@ -290,15 +290,15 @@ class WhisperValues:
290
  "patience": self.patience,
291
  "condition_on_previous_text": self.condition_on_previous_text,
292
  "prompt_reset_on_temperature": self.prompt_reset_on_temperature,
293
- "initial_prompt": self.initial_prompt,
294
  "temperature": self.temperature,
295
  "compression_ratio_threshold": self.compression_ratio_threshold,
296
- "chunk_length_s": self.chunk_length_s,
297
  "batch_size": self.batch_size,
298
  "length_penalty": self.length_penalty,
299
  "repetition_penalty": self.repetition_penalty,
300
  "no_repeat_ngram_size": self.no_repeat_ngram_size,
301
- "prefix": self.prefix,
302
  "suppress_blank": self.suppress_blank,
303
  "suppress_tokens": self.suppress_tokens,
304
  "max_initial_timestamp": self.max_initial_timestamp,
@@ -308,7 +308,7 @@ class WhisperValues:
308
  "max_new_tokens": self.max_new_tokens,
309
  "chunk_length": self.chunk_length,
310
  "hallucination_silence_threshold": self.hallucination_silence_threshold,
311
- "hotwords": self.hotwords,
312
  "language_detection_threshold": self.language_detection_threshold,
313
  "language_detection_segments": self.language_detection_segments,
314
  },
@@ -325,4 +325,4 @@ class WhisperValues:
325
  "hf_token": self.hf_token
326
  }
327
  }
328
- return yaml.dump(data, default_flow_style=False)
 
281
  data = {
282
  "whisper": {
283
  "model_size": self.model_size,
284
+ "lang": "Automatic Detection" if self.lang is None else self.lang,
285
  "is_translate": self.is_translate,
286
  "beam_size": self.beam_size,
287
  "log_prob_threshold": self.log_prob_threshold,
 
290
  "patience": self.patience,
291
  "condition_on_previous_text": self.condition_on_previous_text,
292
  "prompt_reset_on_temperature": self.prompt_reset_on_temperature,
293
+ "initial_prompt": None if not self.initial_prompt else self.initial_prompt,
294
  "temperature": self.temperature,
295
  "compression_ratio_threshold": self.compression_ratio_threshold,
296
+ "chunk_length_s": None if not self.chunk_length_s else self.chunk_length_s,
297
  "batch_size": self.batch_size,
298
  "length_penalty": self.length_penalty,
299
  "repetition_penalty": self.repetition_penalty,
300
  "no_repeat_ngram_size": self.no_repeat_ngram_size,
301
+ "prefix": None if not self.prefix else self.prefix,
302
  "suppress_blank": self.suppress_blank,
303
  "suppress_tokens": self.suppress_tokens,
304
  "max_initial_timestamp": self.max_initial_timestamp,
 
308
  "max_new_tokens": self.max_new_tokens,
309
  "chunk_length": self.chunk_length,
310
  "hallucination_silence_threshold": self.hallucination_silence_threshold,
311
+ "hotwords": None if not self.hotwords else self.hotwords,
312
  "language_detection_threshold": self.language_detection_threshold,
313
  "language_detection_segments": self.language_detection_segments,
314
  },
 
325
  "hf_token": self.hf_token
326
  }
327
  }
328
+ return yaml.dump(data, sort_keys=False, default_flow_style=False)