Update generation_utils.py
Browse files- generation_utils.py +1 -0
generation_utils.py
CHANGED
|
@@ -502,6 +502,7 @@ class DreamGenerationMixin:
|
|
| 502 |
|
| 503 |
# Apply RCR logic if enabled
|
| 504 |
if rcr:
|
|
|
|
| 505 |
self._apply_rcr_logic(x, x0, confidence, mask_index, overtime_confidence,
|
| 506 |
mask_token_id, i, steps, s, t)
|
| 507 |
else:
|
|
|
|
| 502 |
|
| 503 |
# Apply RCR logic if enabled
|
| 504 |
if rcr:
|
| 505 |
+
print(f"[RCR EXEC] Step {i}: RCR logic executed")
|
| 506 |
self._apply_rcr_logic(x, x0, confidence, mask_index, overtime_confidence,
|
| 507 |
mask_token_id, i, steps, s, t)
|
| 508 |
else:
|