Tuananh20015 commited on
Commit
51860ed
·
verified ·
1 Parent(s): 39dbe99

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -52,11 +52,11 @@ def _load_model_cpu(model, filename, *args, **kwargs):
52
  _hub_load.load_model = _load_model_cpu
53
 
54
  # ── Load model once, on GPU (CUDA emulation makes this valid at startup) ───────
55
- print("⏳ Loading VieNeu-TTS v3 Turbo (ONNX CPU Acceleration) ...")
56
  tts = Vieneu(
57
  mode="v3turbo",
58
  device="cpu", # ZeroGPU: keep weights on cuda from the start
59
- backend="onnx", # ONNX is the highly optimized CPU-only path for RAM/CPU speedup
60
  hf_token=os.getenv("HF_TOKEN"),
61
  )
62
  print("✅ Model ready.")
 
52
  _hub_load.load_model = _load_model_cpu
53
 
54
  # ── Load model once, on GPU (CUDA emulation makes this valid at startup) ───────
55
+ print("⏳ Loading VieNeu-TTS v3 Turbo (PyTorch / CUDA) ...")
56
  tts = Vieneu(
57
  mode="v3turbo",
58
  device="cpu", # ZeroGPU: keep weights on cuda from the start
59
+ backend="pytorch", # force the PyTorch engine (ONNX is the CPU-only path)
60
  hf_token=os.getenv("HF_TOKEN"),
61
  )
62
  print("✅ Model ready.")