New image generation img2img. various tweaks and fixes

2024-07-24 04:13:41 -06:00
parent 8d799031cf
commit 80aa2dbb80
9 changed files with 285 additions and 47 deletions
--- a/toolkit/sampler.py
+++ b/toolkit/sampler.py
@@ -41,9 +41,12 @@ sd_config = {
    "prediction_type": "epsilon",
    "sample_max_value": 1.0,
    "set_alpha_to_one": False,
-    "skip_prk_steps": False,
-    "steps_offset": 1,
-    "timestep_spacing": "trailing",
+    # "skip_prk_steps": False,  # for training
+    "skip_prk_steps": True,
+    # "steps_offset": 1,
+    "steps_offset": 0,
+    # "timestep_spacing": "trailing", # for training
+    "timestep_spacing": "leading",
    "trained_betas": None
 }