Update hf_merge.py

hf_merge.py

@@ -58,6 +58,11 @@ class ModelMerger:
 
     def finalize_merge(self, output_dir):
         base_model_path = os.path.join(os.getcwd(), "base_model")
+
+        # Ensure the output directory exists
+        if not os.path.exists(output_dir):
+            os.makedirs(output_dir)
+
         copy_nontensor_files(base_model_path, output_dir)
         save_tensor_map(self.tensor_map, output_dir)