Spaces:
Running
Running
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)
|