From b273f8f2d78cfb6f39f6705bfa917c24d28e2fd3 Mon Sep 17 00:00:00 2001 From: Ethanfel Date: Sun, 5 Apr 2026 10:41:08 +0200 Subject: [PATCH] refactor: remove redundant condition and rename shadowed waveform variable --- nodes/generator.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/nodes/generator.py b/nodes/generator.py index 33c5b22..b534af5 100644 --- a/nodes/generator.py +++ b/nodes/generator.py @@ -38,13 +38,13 @@ class OmniVoiceGenerate: if mode == "voice_design" and not instruct: raise ValueError("voice_design mode requires an instruct string (e.g. 'female, low pitch')") - if mode == "voice_cloning" and ref_audio is not None: + if mode == "voice_cloning": tmp = tempfile.NamedTemporaryFile(suffix=".wav", delete=False) tmp_path = tmp.name tmp.close() try: - waveform = ref_audio["waveform"].squeeze(0).cpu() # (channels, samples) - torchaudio.save(tmp_path, waveform, int(ref_audio["sample_rate"])) + ref_waveform = ref_audio["waveform"].squeeze(0).cpu() # (channels, samples) + torchaudio.save(tmp_path, ref_waveform, int(ref_audio["sample_rate"])) kwargs["ref_audio"] = tmp_path if ref_text: kwargs["ref_text"] = ref_text