Update infer_zipvoice.py

infer_zipvoice.py

@@ -1,58 +1,9 @@
-#!/usr/bin/env python3
-# Copyright         2025  Xiaomi Corp.        (authors: Han Zhu)
-#
-# See ../../../../LICENSE for clarification regarding multiple authors
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-"""
-This script generates speech with our pre-trained ZipVoice or
-    ZipVoice-Distill models. If no local model is specified,
-    Required files will be automatically downloaded from HuggingFace.
-
-Usage:
-
-Note: If you having trouble connecting to HuggingFace,
-    try switching endpoint to mirror site:
-export HF_ENDPOINT=https://hf-mirror.com
-
-(1) Inference of a single sentence:
-
-python3 -m zipvoice.bin.infer_zipvoice \
-    --model-name "zipvoice" \
-    --prompt-wav prompt.wav \
-    --prompt-text "I am a prompt." \
-    --text "I am a sentence." \
-    --res-wav-path result.wav
-
-(2) Inference of a list of sentences:
-
-python3 -m zipvoice.bin.infer_zipvoice \
-    --model-name "zipvoice" \
-    --test-list test.tsv \
-    --res-dir results
-
-`--model-name` can be `zipvoice` or `zipvoice_distill`,
-    which are the models before and after distillation, respectively.
-
-Each line of `test.tsv` is in the format of
-    `{wav_name}\t{prompt_transcription}\t{prompt_wav}\t{text}`.
-"""
-
+# Ẩn mọi GPU để tránh PyTorch cố khởi tạo CUDA khi chạy trên Space CPU
+import os
+os.environ.setdefault("CUDA_VISIBLE_DEVICES", "")
 import argparse
 import datetime as dt
 import json
-import os
 from typing import Optional
 
 import numpy as np