Update tts.py

tts.py

@@ -12,7 +12,8 @@ checkpoint_dir = "model/"
 repo_id = "capleaf/viXTTS"
 use_deepspeed = False
 
-device = "cuda" if torch.cuda.is_available() and "T4" in torch.cuda.get_device_name(0) else "cpu"
+# Kiểm tra xem GPU có sẵn không và sử dụng A100 nếu có
+device = "cuda" if torch.cuda.is_available() and "A100" in torch.cuda.get_device_name(0) else "cpu"
 
 # Tạo thư mục nếu chưa tồn tại
 os.makedirs(checkpoint_dir, exist_ok=True)
@@ -69,10 +70,8 @@ def generate_speech(text, language="vi", speaker_wav=None, normalize_text=True):
     """
     if language not in supported_languages:
         raise ValueError(f"Ngôn ngữ {language} không được hỗ trợ. Chỉ hỗ trợ tiếng Việt (vi) và tiếng Anh (en).")
-
     if len(text) < 2:
         raise ValueError("Văn bản quá ngắn. Vui lòng nhập văn bản dài hơn.")
-
     try:
         # Chuẩn hóa văn bản nếu cần
         if normalize_text and language == "vi":
@@ -86,7 +85,6 @@ def generate_speech(text, language="vi", speaker_wav=None, normalize_text=True):
                 gpt_cond_chunk_len=8 if device == "cuda" else 4,
                 max_ref_length=60 if device == "cuda" else 30,
             )
-
             # Tạo giọng nói
             out = MODEL.inference(
                 text,
@@ -101,8 +99,6 @@ def generate_speech(text, language="vi", speaker_wav=None, normalize_text=True):
         # Lưu file âm thanh
         output_file = "output.wav"
         torchaudio.save(output_file, torch.tensor(out["wav"]).unsqueeze(0).to("cpu"), 24000)
-
         return output_file
-
     except Exception as e:
-        raise RuntimeError(f"Lỗi khi tạo giọng nói: {str(e)}")
+        raise RuntimeError(f"Lỗi khi tạo giọng nói: {str(e)}")