jhj0517 commited on
Commit
e730b1b
·
1 Parent(s): 6b0fe26
Files changed (1) hide show
  1. modules/whisper/data_classes.py +10 -10
modules/whisper/data_classes.py CHANGED
@@ -117,16 +117,16 @@ class DiarizationParams(BaseParams):
117
  label=_("Enable Diarization"),
118
  value=defaults.get("is_diarize", cls.__fields__["is_diarize"].default),
119
  ),
 
 
 
 
 
120
  gr.Textbox(
121
  label=_("HuggingFace Token"),
122
  value=defaults.get("hf_token", cls.__fields__["hf_token"].default),
123
  info=_("This is only needed the first time you download the model")
124
  ),
125
- gr.Dropdown(
126
- label=_("Device"),
127
- choices=["cpu", "cuda"] if available_devices is None else available_devices,
128
- value=defaults.get("device", device),
129
- )
130
  ]
131
 
132
 
@@ -165,17 +165,17 @@ class BGMSeparationParams(BaseParams):
165
  interactive=True,
166
  info=_("Enabling this will remove background music")
167
  ),
168
- gr.Dropdown(
169
- label=_("Device"),
170
- choices=["cpu", "cuda"] if available_devices is None else available_devices,
171
- value=defaults.get("device", device),
172
- ),
173
  gr.Dropdown(
174
  label=_("Model"),
175
  choices=["UVR-MDX-NET-Inst_HQ_4",
176
  "UVR-MDX-NET-Inst_3"] if available_models is None else available_models,
177
  value=defaults.get("model_size", cls.__fields__["model_size"].default),
178
  ),
 
 
 
 
 
179
  gr.Number(
180
  label="Segment Size",
181
  value=defaults.get("segment_size", cls.__fields__["segment_size"].default),
 
117
  label=_("Enable Diarization"),
118
  value=defaults.get("is_diarize", cls.__fields__["is_diarize"].default),
119
  ),
120
+ gr.Dropdown(
121
+ label=_("Device"),
122
+ choices=["cpu", "cuda"] if available_devices is None else available_devices,
123
+ value=defaults.get("device", device),
124
+ ),
125
  gr.Textbox(
126
  label=_("HuggingFace Token"),
127
  value=defaults.get("hf_token", cls.__fields__["hf_token"].default),
128
  info=_("This is only needed the first time you download the model")
129
  ),
 
 
 
 
 
130
  ]
131
 
132
 
 
165
  interactive=True,
166
  info=_("Enabling this will remove background music")
167
  ),
 
 
 
 
 
168
  gr.Dropdown(
169
  label=_("Model"),
170
  choices=["UVR-MDX-NET-Inst_HQ_4",
171
  "UVR-MDX-NET-Inst_3"] if available_models is None else available_models,
172
  value=defaults.get("model_size", cls.__fields__["model_size"].default),
173
  ),
174
+ gr.Dropdown(
175
+ label=_("Device"),
176
+ choices=["cpu", "cuda"] if available_devices is None else available_devices,
177
+ value=defaults.get("device", device),
178
+ ),
179
  gr.Number(
180
  label="Segment Size",
181
  value=defaults.get("segment_size", cls.__fields__["segment_size"].default),