Add Deepseek models into nightly tests (#12865)

scripts/ci/publish_traces.py

@@ -145,7 +145,7 @@ def update_branch_ref(repo_owner, repo_name, branch, commit_sha, token):
     make_github_request(url, token, method="PATCH", data=data)
 
 
-def copy_trace_files(source_dir, target_base_path, is_vlm=False):
+def copy_trace_files(source_dir, target_base_path):
     """Copy trace files and return list of files to upload"""
     files_to_upload = []
 
@@ -171,7 +171,7 @@ def copy_trace_files(source_dir, target_base_path, is_vlm=False):
     return files_to_upload
 
 
-def publish_traces(traces_dir, run_id, run_number, is_vlm=False):
+def publish_traces(traces_dir, run_id, run_number):
     """Publish traces to GitHub repository in a single commit"""
     # Get environment variables
     token = os.getenv("GITHUB_TOKEN")
@@ -186,7 +186,7 @@ def publish_traces(traces_dir, run_id, run_number, is_vlm=False):
     target_base_path = f"traces/{run_id}"
 
     # Copy trace files
-    files_to_upload = copy_trace_files(traces_dir, target_base_path, is_vlm)
+    files_to_upload = copy_trace_files(traces_dir, target_base_path)
 
     if not files_to_upload:
         print("No trace files found to upload")
@@ -261,11 +261,15 @@ def main():
     parser = argparse.ArgumentParser(
         description="Publish performance traces to GitHub repository"
     )
-    parser.add_argument("--vlm", action="store_true", help="Process VLM model traces")
+    parser.add_argument(
+        "--traces-dir",
+        type=str,
+        required=True,
+        help="Traces directory to publish",
+    )
     args = parser.parse_args()
 
     # Get environment variables
-
     run_id = os.getenv("GITHUB_RUN_ID", "test")
     run_number = os.getenv("GITHUB_RUN_NUMBER", "12345")
 
@@ -275,16 +279,12 @@ def main():
         )
         sys.exit(1)
 
-    # Determine traces directory
-    if args.vlm:
-        traces_dir = "performance_profiles_vlms"
-        print("Processing VLM model traces")
-    else:
-        traces_dir = "performance_profiles_text_models"
-        print("Processing text model traces")
+    # Use traces directory
+    traces_dir = args.traces_dir
+    print(f"Processing traces from directory: {traces_dir}")
 
     # Publish traces
-    publish_traces(traces_dir, run_id, run_number, args.vlm)
+    publish_traces(traces_dir, run_id, run_number)
 
 
 if __name__ == "__main__":