refactor: remove unneeded torch.no_grad()

This commit is contained in:
Ayo Ayco 2025-09-03 17:06:11 +02:00
parent 1cb2a50ce5
commit 07672aaf2b
2 changed files with 4 additions and 2 deletions

View file

@ -13,8 +13,6 @@ from transformers.utils import logging
logging.set_verbosity_info()
logger = logging.get_logger(__name__)
torch.no_grad()
class VoiceMapper:
"""Maps speaker names to voice file paths"""

View file

@ -1 +1,5 @@
# defaults to 1.5B
python demo/inference_from_file.py --txt_path demo/text_examples/2p_music.txt --speaker_names Alice Frank
# large -- cannot fit XPU memory (36GB)
python demo/inference_from_file.py --model_path microsoft/VibeVoice-Large --txt_path demo/text_examples/2p_music.txt --speaker_names Alice Frank