minor

Signed-off-by: h-guo18 <67671475+h-guo18@users.noreply.github.com>

Author: h-guo18

examples/speculative_decoding/eagle_utils.py

@@ -395,6 +395,8 @@ def __call__(self, features: list[dict[str, Any]]) -> dict[str, Any]:
 class ARValidationCallback(TrainerCallback):
     def __init__(self, ar_validate_steps: int = 1000):
         self.ar_validate_steps = ar_validate_steps
+        if wandb:
+            wandb.init()
 
     def on_step_end(self, args, state, control, **kwargs):
         if self.ar_validate_steps <= 0: