Update generation_utils.py
Browse files- generation_utils.py +1 -1
generation_utils.py
CHANGED
|
@@ -171,7 +171,7 @@ class DreamGenerationMixin:
|
|
| 171 |
total_steps: int,
|
| 172 |
s: torch.Tensor,
|
| 173 |
t: torch.Tensor,
|
| 174 |
-
ema_beta: float = 0.
|
| 175 |
):
|
| 176 |
"""
|
| 177 |
策略要点(接近 vanilla):
|
|
|
|
| 171 |
total_steps: int,
|
| 172 |
s: torch.Tensor,
|
| 173 |
t: torch.Tensor,
|
| 174 |
+
ema_beta: float = 0.95 # EMA 平滑系数(越大越稳定)
|
| 175 |
):
|
| 176 |
"""
|
| 177 |
策略要点(接近 vanilla):
|