From 07672aaf2b4fb0766270945cab5651df42c9a858 Mon Sep 17 00:00:00 2001
From: Ayo <ayo@ayco.io>
Date: Wed, 3 Sep 2025 17:06:11 +0200
Subject: [PATCH] refactor: remove unneeded torch.no_grad()

---
 demo/inference_from_file.py | 2 --
 example.command             | 4 ++++
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/demo/inference_from_file.py b/demo/inference_from_file.py
index e6e5612..8f67693 100644
--- a/demo/inference_from_file.py
+++ b/demo/inference_from_file.py
@@ -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"""
 
diff --git a/example.command b/example.command
index 805bcbb..a3f36c7 100644
--- a/example.command
+++ b/example.command
@@ -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