r3gm commited on
Commit
6d38b6f
β€’
1 Parent(s): 6984b79

update params

Browse files
Files changed (1) hide show
  1. 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=8,
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
- 4,
2045
  com_t_default,
2046
  "Spanish (es)",
2047
  "English (en)",
2048
  1,
2049
  2,
2050
- "en-CA-ClaraNeural-Female",
2051
- "en-AU-WilliamNeural-Male",
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,