ayo/VibeVoice
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: remove unneeded torch.no_grad()

Browse source
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
Show stats Download patch file Download diff file Expand all files Collapse all files

2
demo/inference_from_file.py
View file

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

4
example.command
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 python demo/inference_from_file.py --model_path microsoft/VibeVoice-Large --txt_path demo/text_examples/2p_music.txt --speaker_names Alice Frank