diff --git a/src/transformers/utils/notebook.py b/src/transformers/utils/notebook.py index 9704aca242a60f..300ff6fde48238 100644 --- a/src/transformers/utils/notebook.py +++ b/src/transformers/utils/notebook.py @@ -114,6 +114,8 @@ def __init__( self.last_value = None self.comment = None self.output = None + self.value = None + self.label = None def update(self, value: int, force_update: bool = False, comment: str = None): """