Add Positive and Negative Aesthetic Score to the parameters #8

2023-11-22 00:51:02 +08:00
parent 07ecca0de4
commit 2950c76e9d
1 changed files with 9 additions and 0 deletions
@@ -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"
        )