From 27827985e3fb0d3ba2690c0f9fa8958a7c848fa0 Mon Sep 17 00:00:00 2001
From: wailovet <wailovet@163.com>
Date: Sun, 30 Jun 2024 23:13:08 +0800
Subject: [PATCH] 'update'

---
 mz_train_tools_core_HYDiT.py | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/mz_train_tools_core_HYDiT.py b/mz_train_tools_core_HYDiT.py
index 2a3f4d4..04dacf5 100644
--- a/mz_train_tools_core_HYDiT.py
+++ b/mz_train_tools_core_HYDiT.py
@@ -876,12 +876,8 @@ def MZ_HYDiTSimpleT2I_call(args={}):
         'scheduler': 'DPMSolverMultistepScheduler',
         'name': 'DPMSolverMultistepScheduler',
         'kwargs': {
-            'beta_schedule': 'scaled_linear',
-            'beta_start': 0.00085,
-            'beta_end': 0.03,
-            'prediction_type': 'v_prediction',
-            'trained_betas': None,
-            'solver_order': 2,
+            'beta_schedule': 'scaled_linear', 
+            'prediction_type': 'v_prediction', 
             'algorithm_type': 'dpmsolver++',
             "use_karras_sigmas": True,
         }