Upload files with `vila-upload`.

modeling_vila.py

@@ -350,7 +350,7 @@ class VILAPretrainedModel(PreTrainedModel):
                         os.symlink(full_file_name, os.path.join(output_dir, file_name))
                         print("[HF remote code] linking", full_file_name, "to", output_dir)
 
-    def save_pretrained(self, output_dir, state_dict=None):
+    def save_pretrained(self, output_dir, state_dict=None, **kwargs):
         if state_dict is None:
             # other wise fetch from deepspeed
             # state_dict = accelerator.get_state_dict(is_deepspeed_enabled)
@@ -590,12 +590,11 @@ class VILAForCasualLM(VILAPretrainedModel):
 
     def train(self, mode: bool = True):
         if mode:
-            print(f"Set padding side to right for training, {mode=}")
             self.tokenizer.padding_side = "right"
         else:
-            print(f"Set padding side to left for evaluation, {mode=}")
             self.tokenizer.padding_side = "left"
         super().train(mode)
+        return self
 
     def _embed(
         self,