Update trainer.py

trainer.py

@@ -97,7 +97,7 @@ class Trainer:
         output_dir.mkdir(parents=True)
 
         if upload_to_hub:
-            self.join_model_library_org()
+            self.join_model_library_org(self.hf_token if self.hf_token else input_token)
 
         config = OmegaConf.load('Tune-A-Video/configs/man-surfing.yaml')
         config.pretrained_model_path = self.download_base_model(base_model)
@@ -154,7 +154,7 @@ class Trainer:
         if remove_gpu_after_training:
             space_id = os.getenv('SPACE_ID')
             if space_id:
-                api = HfApi(token=hf_token if self.hf_token else input_token)
+                api = HfApi(token=self.hf_token if self.hf_token else input_token)
                 api.request_space_hardware(repo_id=space_id,
                                            hardware='cpu-basic')