Update _upload_file

app.py

@@ -121,17 +121,17 @@ class HuggingFaceModelProcessor:
         """Extract model name from model ID."""
         return model_id.split('/')[-1]
 
-    def _upload_file(self, token: str, path_or_fileobj: str, path_in_repo: str, repo_id: str) -> None:
+    def _upload_file(self, processing_config: ModelProcessingConfig, path_or_fileobj: str, path_in_repo: str) -> None:
         """Upload a file to Hugging Face repository."""
         if self.RUN_LOCALLY == "1":
             print("Skipping upload...")
             return
 
-        api = HfApi(token=token)
+        api = HfApi(token=processing_config.token)
         api.upload_file(
             path_or_fileobj=path_or_fileobj,
             path_in_repo=path_in_repo,
-            repo_id=repo_id,
+            repo_id=processing_config.new_repo_id,
         )
 
     def _generate_importance_matrix(self, model_path: str, train_data_path: str, output_path: str) -> None:
@@ -409,7 +409,7 @@ llama-server --hf-repo "{new_repo_id}" --hf-file "{gguf_name}" -c 4096
         else:
             try:
                 print(f"Uploading quantized model: {os.path.abspath(quant_config.quantized_gguf)}")
-                self._upload_file(processing_config.token, quant_config.quantized_gguf, output_config.filename, processing_config.new_repo_id)
+                self._upload_file(processing_config, quant_config.quantized_gguf, output_config.filename)
             except Exception as e:
                 raise GGUFConverterError(f"Error uploading quantized model: {e}")
 
@@ -417,13 +417,13 @@ llama-server --hf-repo "{new_repo_id}" --hf-file "{gguf_name}" -c 4096
         if quant_config.use_imatrix and os.path.isfile(quant_config.imatrix_file):
             try:
                 print(f"Uploading imatrix.dat: {os.path.abspath(quant_config.imatrix_file)}")
-                self._upload_file(processing_config.token, quant_config.imatrix_file, f"{processing_config.model_name}-imatrix.dat", processing_config.new_repo_id)
+                self._upload_file(processing_config, quant_config.imatrix_file, f"{processing_config.model_name}-imatrix.dat")
             except Exception as e:
                 raise GGUFConverterError(f"Error uploading imatrix.dat: {e}")
 
         # Upload README.md
         readme_path = self._generate_readme(processing_config.outdir, processing_config.token, processing_config.model_id, processing_config.new_repo_id, output_config.filename)
-        self._upload_file(processing_config.token, readme_path, "README.md", processing_config.new_repo_id)
+        self._upload_file(processing_config, readme_path, "README.md")
 
         print(f"Uploaded successfully with {quant_config.imatrix_method if quant_config.use_imatrix else quant_config.method} option!")