Upload export_all.py with huggingface_hub

export_all.py

@@ -35,6 +35,12 @@ def main():
         default="onnx_models",
         help="Output directory for ONNX models",
     )
+    parser.add_argument(
+        "--model",
+        type=str,
+        default="facebook/sam-audio-small",
+        help="SAM-Audio model ID (e.g., facebook/sam-audio-small, facebook/sam-audio-large, facebook/sam-audio-base-tv)",
+    )
     parser.add_argument(
         "--verify",
         action="store_true",
@@ -69,28 +75,28 @@ def main():
     
     # Export DACVAE
     if not args.skip_dacvae:
-        export_args = ["--output-dir", args.output_dir]
+        export_args = ["--output-dir", args.output_dir, "--model-id", args.model]
         if args.verify:
             export_args.append("--verify")
         results["DACVAE"] = run_export("onnx_export.export_dacvae", export_args)
     
     # Export T5
     if not args.skip_t5:
-        export_args = ["--output-dir", args.output_dir]
+        export_args = ["--output-dir", args.output_dir, "--model-id", args.model]
         if args.verify:
             export_args.append("--verify")
         results["T5"] = run_export("onnx_export.export_t5", export_args)
     
     # Export DiT
     if not args.skip_dit:
-        export_args = ["--output-dir", args.output_dir]
+        export_args = ["--output-dir", args.output_dir, "--model-id", args.model]
         if args.verify:
             export_args.append("--verify")
         results["DiT"] = run_export("onnx_export.export_dit", export_args)
     
     # Export Vision Encoder
     if not args.skip_vision:
-        export_args = ["--output", args.output_dir]
+        export_args = ["--output", args.output_dir, "--model", args.model]
         results["Vision"] = run_export("onnx_export.export_vision", export_args)
     
     # Print summary