Update generation_utils.py

generation_utils.py

@@ -192,7 +192,7 @@ class DreamGenerationMixin:
 
             # 目标累计约束：截至本步应累计的生成数
             if step < total_steps - 1:
-                target_cum = int(num_mask_token * (1 - s / t))
+                target_cum = int(num_mask_token * (1 - s))  # 累计目标：随 s 递减而线性增长
                 gen_mask = overtime_confidence[j] > 0
                 current_gen = int(gen_mask.sum().item())
                 # 若超额，则按最低置信度回遮