From f6ea252652eea99e8056d1497b528853cdb37686 Mon Sep 17 00:00:00 2001 From: bennykok Date: Sat, 10 Aug 2024 10:35:48 -0700 Subject: [PATCH] fix: log when random seed is applied --- custom_routes.py | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/custom_routes.py b/custom_routes.py index 7703ec2..362b159 100644 --- a/custom_routes.py +++ b/custom_routes.py @@ -221,18 +221,23 @@ def apply_random_seed_to_workflow(workflow_api): if isinstance(workflow_api[key]['inputs']['seed'], list): continue if workflow_api[key]['class_type'] == "PromptExpansion": - workflow_api[key]['inputs']['seed'] = randomSeed(8); + workflow_api[key]['inputs']['seed'] = randomSeed(8) + logger.info(f"Applied random seed {workflow_api[key]['inputs']['seed']} to PromptExpansion") continue if workflow_api[key]['class_type'] == "RandomNoise": - workflow_api[key]['inputs']['noise_seed'] = randomSeed(); + workflow_api[key]['inputs']['noise_seed'] = randomSeed() + logger.info(f"Applied random noise_seed {workflow_api[key]['inputs']['noise_seed']} to RandomNoise") continue if workflow_api[key]['class_type'] == "KSamplerAdvanced": - workflow_api[key]['inputs']['noise_seed'] = randomSeed(); + workflow_api[key]['inputs']['noise_seed'] = randomSeed() + logger.info(f"Applied random noise_seed {workflow_api[key]['inputs']['noise_seed']} to KSamplerAdvanced") continue if workflow_api[key]['class_type'] == "SamplerCustom": - workflow_api[key]['inputs']['noise_seed'] = randomSeed(); + workflow_api[key]['inputs']['noise_seed'] = randomSeed() + logger.info(f"Applied random noise_seed {workflow_api[key]['inputs']['noise_seed']} to SamplerCustom") continue - workflow_api[key]['inputs']['seed'] = randomSeed(); + workflow_api[key]['inputs']['seed'] = randomSeed() + logger.info(f"Applied random seed {workflow_api[key]['inputs']['seed']} to {workflow_api[key]['class_type']}") def apply_inputs_to_workflow(workflow_api: Any, inputs: Any, sid: str = None): # Loop through each of the inputs and replace them