Update generation_utils.py
Browse files- generation_utils.py +1 -0
generation_utils.py
CHANGED
|
@@ -458,6 +458,7 @@ class DreamGenerationMixin:
|
|
| 458 |
|
| 459 |
if rcr:
|
| 460 |
# === 仅在 rcr=True 时：应用 RCR；不会触碰 baseline 分支实现 ===
|
|
|
|
| 461 |
self._apply_rcr_logic(
|
| 462 |
x, x0, confidence, mask_index, overtime_confidence,
|
| 463 |
mask_token_id, i, steps, s, t
|
|
|
|
| 458 |
|
| 459 |
if rcr:
|
| 460 |
# === 仅在 rcr=True 时：应用 RCR；不会触碰 baseline 分支实现 ===
|
| 461 |
+
print("rcr")
|
| 462 |
self._apply_rcr_logic(
|
| 463 |
x, x0, confidence, mask_index, overtime_confidence,
|
| 464 |
mask_token_id, i, steps, s, t
|