lj1995 commited on
Commit
a010f18
Β·
1 Parent(s): 3c7d6a8

Update inference_webui.py

Browse files
Files changed (1) hide show
  1. inference_webui.py +14 -0
inference_webui.py CHANGED
@@ -343,6 +343,20 @@ def merge_short_text_in_array(texts, threshold):
343
  cache= {}
344
  def get_tts_wav(ref_wav_path, prompt_text, prompt_language, text, text_language, how_to_cut=i18n("δΈεˆ‡"), top_k=20, top_p=0.6, temperature=0.6, ref_free = False,speed=1,if_freeze=False,inp_refs=123):
345
  global cache
 
 
 
 
 
 
 
 
 
 
 
 
 
 
346
  try:
347
  if ref_wav_path:pass
348
  else:gr.Warning(i18n('θ―·δΈŠδΌ ε‚θ€ƒιŸ³ι’‘'))
 
343
  cache= {}
344
  def get_tts_wav(ref_wav_path, prompt_text, prompt_language, text, text_language, how_to_cut=i18n("δΈεˆ‡"), top_k=20, top_p=0.6, temperature=0.6, ref_free = False,speed=1,if_freeze=False,inp_refs=123):
345
  global cache
346
+
347
+ import psutil
348
+
349
+ # θŽ·ε–ε†…ε­˜δΏ‘ζ―
350
+ memory_info = psutil.virtual_memory()
351
+
352
+ # ζ‰“ε°ζ€»ε†…ε­˜ε’Œε‰©δ½™ε†…ε­˜
353
+ total_memory = memory_info.total / (1024 ** 3) # 转捒为 GB
354
+ available_memory = memory_info.available / (1024 ** 3) # 转捒为 GB
355
+
356
+ print(f"ζ€»ε†…ε­˜: {total_memory:.2f} GB")
357
+ print(f"ε‰©δ½™ε†…ε­˜: {available_memory:.2f} GB")
358
+
359
+
360
  try:
361
  if ref_wav_path:pass
362
  else:gr.Warning(i18n('θ―·δΈŠδΌ ε‚θ€ƒιŸ³ι’‘'))