jhj0517 commited on
Commit
ac4bff9
·
1 Parent(s): 1599c90

add `condition_on_previous_text`

Browse files
Files changed (1) hide show
  1. modules/whisper_data_class.py +6 -0
modules/whisper_data_class.py CHANGED
@@ -13,6 +13,7 @@ class WhisperGradioComponents:
13
  compute_type: gr.Dropdown
14
  best_of: gr.Number
15
  patience: gr.Number
 
16
  """
17
  A data class to pass Gradio components to the function before Gradio pre-processing.
18
  See this documentation for more information about Gradio pre-processing: https://www.gradio.app/docs/components
@@ -41,6 +42,10 @@ class WhisperGradioComponents:
41
  Number of candidates when sampling with non-zero temperature.
42
  patience: gr.Number
43
  Beam search patience factor.
 
 
 
 
44
  """
45
 
46
  def to_list(self) -> list:
@@ -66,6 +71,7 @@ class WhisperValues:
66
  compute_type: str
67
  best_of: int
68
  patience: float
 
69
  """
70
  A data class to use Whisper parameters in your function after Gradio pre-processing.
71
  See this documentation for more information about Gradio pre-processing: : https://www.gradio.app/docs/components
 
13
  compute_type: gr.Dropdown
14
  best_of: gr.Number
15
  patience: gr.Number
16
+ condition_on_previous_text: gr.Checkbox
17
  """
18
  A data class to pass Gradio components to the function before Gradio pre-processing.
19
  See this documentation for more information about Gradio pre-processing: https://www.gradio.app/docs/components
 
42
  Number of candidates when sampling with non-zero temperature.
43
  patience: gr.Number
44
  Beam search patience factor.
45
+ condition_on_previous_text: bool
46
+ if True, the previous output of the model is provided as a prompt for the next window;
47
+ disabling may make the text inconsistent across windows, but the model becomes less prone to
48
+ getting stuck in a failure loop, such as repetition looping or timestamps going out of sync.
49
  """
50
 
51
  def to_list(self) -> list:
 
71
  compute_type: str
72
  best_of: int
73
  patience: float
74
+ condition_on_previous_text: bool
75
  """
76
  A data class to use Whisper parameters in your function after Gradio pre-processing.
77
  See this documentation for more information about Gradio pre-processing: : https://www.gradio.app/docs/components