refactor: remove unneeded torch.no_grad()
This commit is contained in:
parent
1cb2a50ce5
commit
07672aaf2b
2 changed files with 4 additions and 2 deletions
|
@ -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"""
|
||||
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in a new issue