blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-10-06 06:29:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

Compatible with older graphics cards

Browse Source 
Test equipment: rtx1060
This commit is contained in:
刘悦 2025-01-26 11:03:21 +08:00 committed by GitHub
parent b7a904a671
commit 822449c479
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
GPT_SoVITS/inference_webui.py
View File

@ -73,7 +73,15 @@ is_share = os.environ.get("is_share", "False")
is_share = eval(is_share)
if "_CUDA_VISIBLE_DEVICES" in os.environ:
os.environ["CUDA_VISIBLE_DEVICES"] = os.environ["_CUDA_VISIBLE_DEVICES"]
is_half = eval(os.environ.get("is_half", "True")) and torch.cuda.is_available()
is_fp16_supported = torch.cuda.is_available() and torch.cuda.get_device_capability()[0] > 7
if is_fp16_supported:
is_half = True
else:
is_half = False
punctuation = set(['!', '?', '…', ',', '.', '-'," "])
import gradio as gr
from transformers import AutoModelForMaskedLM, AutoTokenizer