From 2950c76e9d73f885e65d2ac6a109a420e7bf3591 Mon Sep 17 00:00:00 2001 From: Rhys Yang Date: Tue, 21 Nov 2023 23:03:53 +0800 Subject: [PATCH] Add Positive and Negative Aesthetic Score to the `parameters` #8 --- nodes.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/nodes.py b/nodes.py index e7c28c7..05b6554 100644 --- a/nodes.py +++ b/nodes.py @@ -620,6 +620,14 @@ class SDParameterGenerator: base_steps = int(steps * refiner_start) refiner_steps = steps - base_steps + if model_version == "SDXL 1024px": + ascore = ( + f"Positive aesthetic score: {positive_ascore},\n" + f"Negative aesthetic score: {negative_ascore},\n" + ) + else: + ascore = "" + parameters = ( f"Model: {ckpt_name},\n" f"Seed: {str(seed)},\n" @@ -627,6 +635,7 @@ class SDParameterGenerator: f"CFG scale: {str(cfg)},\n" f"Sampler: {sampler_name},\n" f"Scheduler: {scheduler},\n" + f"{ascore}" f"Size: {str(width)}x{str(height)},\n" f"Batch size: {str(batch_size)}\n" )