'update'

2024-06-30 23:19:17 +08:00
parent 9e15bb4cf1
commit 2b1adb20a8
1 changed files with 5 additions and 1 deletions
@@ -878,7 +878,11 @@ def MZ_HYDiTSimpleT2I_call(args={}):
        'kwargs': {
            'beta_schedule': 'scaled_linear',
            'beta_start': 0.00085,
-            'beta_end': 0.03, 
+            'beta_end': 0.03,
+            'prediction_type': 'v_prediction',
+            'trained_betas': None,
+            'solver_order': 2,
+            'algorithm_type': 'dpmsolver++',
            "use_karras_sigmas": True,
        }
    }