From 9dcf692e7d5d044ae837f19a795ca25b37a6a137 Mon Sep 17 00:00:00 2001 From: c0008 <139379535+c0008@users.noreply.github.com> Date: Tue, 15 Apr 2025 19:19:01 +0200 Subject: [PATCH] generate a random seed when no explicit seed is provided --- tangoflux/model.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tangoflux/model.py b/tangoflux/model.py index 8a13706..a3f5ce0 100644 --- a/tangoflux/model.py +++ b/tangoflux/model.py @@ -286,13 +286,16 @@ def inference_flow( timesteps=None, guidance_scale=3, duration=10, - seed=0, + seed=None, disable_progress=False, num_samples_per_prompt=1, callback_on_step_end=None, ): """Only tested for single inference. Haven't test for batch inference""" + if seed is None: + seed = random.randint(0, 2**32 - 1) # Generate a random integer as seed + torch.manual_seed(seed) if torch.cuda.is_available(): torch.cuda.manual_seed(seed)