Spaces:
Running
Running
update params
Browse files- app_rvc.py +6 -4
app_rvc.py
CHANGED
|
@@ -284,6 +284,7 @@ class SoniTranslate(SoniTrCache):
|
|
| 284 |
)
|
| 285 |
|
| 286 |
self.device = os.environ.get("SONITR_DEVICE")
|
|
|
|
| 287 |
self.result_diarize = None
|
| 288 |
self.align_language = None
|
| 289 |
self.result_source_lang = None
|
|
@@ -1886,10 +1887,11 @@ def create_gui(theme, logs_in_gui=False):
|
|
| 1886 |
label=lg_conf["ctype_label"],
|
| 1887 |
info=lg_conf["ctype_info"],
|
| 1888 |
)
|
|
|
|
| 1889 |
batch_size = gr.Slider(
|
| 1890 |
minimum=1,
|
| 1891 |
maximum=32,
|
| 1892 |
-
value=
|
| 1893 |
label=lg_conf["batchz_label"],
|
| 1894 |
info=lg_conf["batchz_info"],
|
| 1895 |
step=1,
|
|
@@ -2041,14 +2043,14 @@ def create_gui(theme, logs_in_gui=False):
|
|
| 2041 |
"",
|
| 2042 |
False,
|
| 2043 |
whisper_model_default,
|
| 2044 |
-
|
| 2045 |
com_t_default,
|
| 2046 |
"Spanish (es)",
|
| 2047 |
"English (en)",
|
| 2048 |
1,
|
| 2049 |
2,
|
| 2050 |
-
"en-
|
| 2051 |
-
"en-
|
| 2052 |
],
|
| 2053 |
], # no update
|
| 2054 |
fn=SoniTr.batch_multilingual_media_conversion,
|
|
|
|
| 284 |
)
|
| 285 |
|
| 286 |
self.device = os.environ.get("SONITR_DEVICE")
|
| 287 |
+
self.device = self.device if os.environ.get("ZERO_GPU") != "TRUE" else "cuda"
|
| 288 |
self.result_diarize = None
|
| 289 |
self.align_language = None
|
| 290 |
self.result_source_lang = None
|
|
|
|
| 1887 |
label=lg_conf["ctype_label"],
|
| 1888 |
info=lg_conf["ctype_info"],
|
| 1889 |
)
|
| 1890 |
+
batch_size_value = 8 if os.environ.get("ZERO_GPU") != "TRUE" else 32
|
| 1891 |
batch_size = gr.Slider(
|
| 1892 |
minimum=1,
|
| 1893 |
maximum=32,
|
| 1894 |
+
value=batch_size_value,
|
| 1895 |
label=lg_conf["batchz_label"],
|
| 1896 |
info=lg_conf["batchz_info"],
|
| 1897 |
step=1,
|
|
|
|
| 2043 |
"",
|
| 2044 |
False,
|
| 2045 |
whisper_model_default,
|
| 2046 |
+
batch_size_value,
|
| 2047 |
com_t_default,
|
| 2048 |
"Spanish (es)",
|
| 2049 |
"English (en)",
|
| 2050 |
1,
|
| 2051 |
2,
|
| 2052 |
+
"en-US-EmmaMultilingualNeural-Female",
|
| 2053 |
+
"en-US-AndrewMultilingualNeural-Male",
|
| 2054 |
],
|
| 2055 |
], # no update
|
| 2056 |
fn=SoniTr.batch_multilingual_media_conversion,
|