Add revision to push_to_hub (#292)

* Add revision to push_to_hub

* make style

* remove fairscale

* style happy

* Update modeling_base.py

* Update modeling_decoder.py

optimum/neuron/modeling_base.py

@@ -317,6 +317,7 @@ def push_to_hub(
         save_directory: str,
         repository_id: str,
         private: Optional[bool] = None,
+        revision: Optional[str] = None,
         use_auth_token: Union[bool, str] = True,
         endpoint: Optional[str] = None,
     ) -> str:
@@ -348,6 +349,7 @@ def push_to_hub(
                     repo_id=repository_id,
                     path_or_fileobj=os.path.join(os.getcwd(), local_file_path),
                     path_in_repo=hub_file_path,
+                    revision=revision,
                 )
 
     def forward(self, *args, **kwargs):

optimum/neuron/modeling_decoder.py

@@ -259,6 +259,7 @@ def push_to_hub(
         save_directory: str,
         repository_id: str,
         private: Optional[bool] = None,
+        revision: Optional[str] = None,
         use_auth_token: Union[bool, str] = True,
         endpoint: Optional[str] = None,
     ) -> str:
@@ -290,4 +291,5 @@ def push_to_hub(
                     repo_id=repository_id,
                     path_or_fileobj=os.path.join(os.getcwd(), local_file_path),
                     path_in_repo=hub_file_path,
+                    revision=revision,
                 )

optimum/neuron/trainers.py

@@ -27,8 +27,6 @@
 import torch
 from packaging import version
 from transformers import PreTrainedModel, Seq2SeqTrainer, Trainer, TrainingArguments
-from transformers.dependency_versions_check import dep_version_check
-from transformers.integrations import is_fairscale_available
 from transformers.modeling_utils import unwrap_model
 from transformers.trainer import (
     OPTIMIZER_NAME,
@@ -80,9 +78,6 @@
 else:
     IS_SAGEMAKER_MP_POST_1_10 = False
 
-if is_fairscale_available():
-    dep_version_check("fairscale")
-
 
 logger = logging.get_logger("transformers.trainer")