fix typo in method name

Lightning-Universe · Jun 23, 2022 · f48ba80 · f48ba80
1 parent 4a14a94
commit f48ba80
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lightning_transformers/core/model.py b/lightning_transformers/core/model.py
@@ -58,12 +58,12 @@ def __init__(
         self.load_weights = load_weights
         self.model_data_kwargs = model_data_kwargs
         self.downstream_model_type = downstream_model_type
-        self.intiailaize_model(pretrained_model_name_or_path)
+        self.initialize_model(pretrained_model_name_or_path)
         self._tokenizer = tokenizer  # necessary for hf_pipeline
         self._hf_pipeline = None
         self._hf_pipeline_kwargs = pipeline_kwargs or {}
 
-    def intiailaize_model(self, pretrained_model_name_or_path: str):
+    def initialize_model(self, pretrained_model_name_or_path: str):
         """create and initialize the model to use with this task,
 
         Feel free to overwrite this method if you are initializing the model in a different way