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,
|