From d700b56c9c402058265ecce1d2dd4563e6dcdca6 Mon Sep 17 00:00:00 2001 From: Kirill Date: Sat, 7 Jun 2025 23:36:49 +0200 Subject: [PATCH] docs: add missing torch import to Python usage example in README --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index b453b4bd4..2b1addae2 100644 --- a/README.md +++ b/README.md @@ -189,7 +189,7 @@ result = model.transcribe(audio, batch_size=batch_size) print(result["segments"]) # before alignment # delete model if low on GPU resources -# import gc; gc.collect(); torch.cuda.empty_cache(); del model +# import gc; import torch; gc.collect(); torch.cuda.empty_cache(); del model # 2. Align whisper output model_a, metadata = whisperx.load_align_model(language_code=result["language"], device=device) @@ -198,7 +198,7 @@ result = whisperx.align(result["segments"], model_a, metadata, audio, device, re print(result["segments"]) # after alignment # delete model if low on GPU resources -# import gc; gc.collect(); torch.cuda.empty_cache(); del model_a +# import gc; import torch; gc.collect(); torch.cuda.empty_cache(); del model_a # 3. Assign speaker labels diarize_model = whisperx.diarize.DiarizationPipeline(use_auth_token=YOUR_HF_TOKEN, device=device)