Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+evaluation/build/lib.linux-x86_64-cpython-313/utils/benchmark_utils.cpython-313-x86_64-linux-gnu.so filter=lfs diff=lfs merge=lfs -text
+evaluation/build/temp.linux-x86_64-cpython-313/utils/benchmark_utils.o filter=lfs diff=lfs merge=lfs -text
+evaluation/utils/benchmark_utils.cpython-313-x86_64-linux-gnu.so filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,8 @@
+# Python bytecode
+*.pyc
+*.pyo
+__pycache__/
+.env
+*.log
+build/
+.swp

README.md

@@ -0,0 +1,125 @@
+---
+license: mit
+library_name: transformers
+---
+# MyAwesomeModel
+<!-- markdownlint-disable first-line-h1 -->
+<!-- markdownlint-disable html -->
+<!-- markdownlint-disable no-duplicate-header -->
+
+<div align="center">
+  <img src="figures/fig1.png" width="60%" alt="MyAwesomeModel" />
+</div>
+<hr>
+
+<div align="center" style="line-height: 1;">
+  <a href="LICENSE" style="margin: 2px;">
+    <img alt="License" src="figures/fig2.png" style="display: inline-block; vertical-align: middle;"/>
+  </a>
+</div>
+
+## 1. Introduction
+
+The MyAwesomeModel has undergone a significant version upgrade. In the latest update, MyAwesomeModel has significantly improved its depth of reasoning and inference capabilities by leveraging increased computational resources and introducing algorithmic optimization mechanisms during post-training. The model has demonstrated outstanding performance across various benchmark evaluations, including mathematics, programming, and general logic. Its overall performance is now approaching that of other leading models.
+
+<p align="center">
+  <img width="80%" src="figures/fig3.png">
+</p>
+
+Compared to the previous version, the upgraded model shows significant improvements in handling complex reasoning tasks. For instance, in the AIME 2025 test, the model’s accuracy has increased from 70% in the previous version to 87.5% in the current version. This advancement stems from enhanced thinking depth during the reasoning process: in the AIME test set, the previous model used an average of 12K tokens per question, whereas the new version averages 23K tokens per question.
+
+Beyond its improved reasoning capabilities, this version also offers a reduced hallucination rate and enhanced support for function calling.
+
+## 2. Evaluation Results
+
+### Comprehensive Benchmark Results
+
+<div align="center">
+
+| | Benchmark | Model1 | Model2 | Model1-v2 | MyAwesomeModel |
+|---|---|---|---|---|---|
+| **Core Reasoning Tasks** | Math Reasoning | 0.510 | 0.535 | 0.521 | 0.550 |
+| | Logical Reasoning | 0.789 | 0.801 | 0.810 | 0.819 |
+| | Common Sense | 0.716 | 0.702 | 0.725 | 0.736 |
+| **Language Understanding** | Reading Comprehension | 0.671 | 0.685 | 0.690 | 0.700 |
+| | Question Answering | 0.582 | 0.599 | 0.601 | 0.607 |
+| | Text Classification | 0.803 | 0.811 | 0.820 | 0.828 |
+| | Sentiment Analysis | 0.777 | 0.781 | 0.790 | 0.792 |
+| **Generation Tasks** | Code Generation | 0.615 | 0.631 | 0.640 | 0.650 |
+| | Creative Writing | 0.588 | 0.579 | 0.601 | 0.650 |
+| | Dialogue Generation | 0.621 | 0.635 | 0.639 | 0.644 |
+| | Summarization | 0.745 | 0.755 | 0.760 | 0.767 |
+| **Specialized Capabilities**| Translation | 0.782 | 0.799 | 0.801 | 0.600 |
+| | Knowledge Retrieval | 0.651 | 0.668 | 0.670 | 0.650 |
+| | Instruction Following | 0.733 | 0.749 | 0.751 | 0.778 |
+| | Safety Evaluation | 0.718 | 0.701 | 0.725 | 0.610 |
+
+</div>
+
+### Overall Performance Summary
+The MyAwesomeModel demonstrates strong performance across all evaluated benchmark categories, with particularly notable results in reasoning and generation tasks.
+
+## 3. Chat Website & API Platform
+We offer a chat interface and API for you to interact with MyAwesomeModel. Please check our official website for more details.
+
+## 4. How to Run Locally
+
+Please refer to our code repository for more information about running MyAwesomeModel locally.
+
+Compared to previous versions, the usage recommendations for MyAwesomeModel have the following changes:
+
+1. System prompt is supported.
+2. It is not required to add special tokens at the beginning of the output to force the model into a specific thinking pattern.
+
+The model architecture of MyAwesomeModel-Small is identical to its base model, but it shares the same tokenizer configuration as the main MyAwesomeModel. This model can be run in the same manner as its base model.
+
+### System Prompt
+We recommend using the following system prompt with a specific date.
+```
+You are MyAwesomeModel, a helpful AI assistant.
+Today is {current date}.
+```
+For example,
+```
+You are MyAwesomeModel, a helpful AI assistant.
+Today is May 28, 2025, Monday.
+```
+### Temperature
+We recommend setting the temperature parameter $T_{model}$ to 0.6. 
+
+### Prompts for File Uploading and Web Search
+For file uploading, please follow the template to create prompts, where {file_name}, {file_content} and {question} are arguments.
+```
+file_template = \
+"""[file name]: {file_name}
+[file content begin]
+{file_content}
+[file content end]
+{question}"""
+```
+For web search enhanced generation, we recommend the following prompt template where {search_results}, {cur_date}, and {question} are arguments.
+```
+search_answer_en_template = \
+'''# The following contents are the search results related to the user's message:
+{search_results}
+In the search results I provide to you, each result is formatted as [webpage X begin]...[webpage X end], where X represents the numerical index of each article. Please cite the context at the end of the relevant sentence when appropriate. Use the citation format [citation:X] in the corresponding part of your answer. If a sentence is derived from multiple contexts, list all relevant citation numbers, such as [citation:3][citation:5]. Be sure not to cluster all citations at the end; instead, include them in the corresponding parts of the answer.
+When responding, please keep the following points in mind:
+- Today is {cur_date}.
+- Not all content in the search results is closely related to the user's question. You need to evaluate and filter the search results based on the question.
+- For listing-type questions (e.g., listing all flight information), try to limit the answer to 10 key points and inform the user that they can refer to the search sources for complete information. Prioritize providing the most complete and relevant items in the list. Avoid mentioning content not provided in the search results unless necessary.
+- For creative tasks (e.g., writing an essay), ensure that references are cited within the body of the text, such as [citation:3][citation:5], rather than only at the end of the text. You need to interpret and summarize the user's requirements, choose an appropriate format, fully utilize the search results, extract key information, and generate an answer that is insightful, creative, and professional. Extend the length of your response as much as possible, addressing each point in detail and from multiple perspectives, ensuring the content is rich and thorough.
+- If the response is lengthy, structure it well and summarize it in paragraphs. If a point-by-point format is needed, try to limit it to 5 points and merge related content.
+- For objective Q&A, if the answer is very brief, you may add one or two related sentences to enrich the content.
+- Choose an appropriate and visually appealing format for your response based on the user's requirements and the content of the answer, ensuring strong readability.
+- Your answer should synthesize information from multiple relevant webpages and avoid repeatedly citing the same webpage.
+- Unless the user requests otherwise, your response should be in the same language as the user's question.
+# The user's message is:
+{question}'''
+```
+
+## 5. License
+This code repository is licensed under the [MIT License](LICENSE). The use of MyAwesomeModel models is also subject to the [MIT License](LICENSE). The model series supports commercial use and distillation.
+
+## 6. Contact
+If you have any questions, please raise an issue on our GitHub repository or contact us at contact@MyAwesomeModel.ai.
+```

checkpoints/step_100/config.json

@@ -0,0 +1,4 @@
+{
+    "model_type": "bert",
+    "architectures": ["BertModel"]
+  }

checkpoints/step_100/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:965362299a238de576a92dfdd3e32aea7a2bacc94b2c41541c8c9258b923f587
+size 23

checkpoints/step_1000/config.json

@@ -0,0 +1,4 @@
+{
+    "model_type": "bert",
+    "architectures": ["BertModel"]
+  }

checkpoints/step_1000/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:965362299a238de576a92dfdd3e32aea7a2bacc94b2c41541c8c9258b923f587
+size 23

checkpoints/step_200/config.json

@@ -0,0 +1,4 @@
+{
+  "model_type": "bert",
+  "architectures": ["BertModel"]
+}

checkpoints/step_200/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:965362299a238de576a92dfdd3e32aea7a2bacc94b2c41541c8c9258b923f587
+size 23

checkpoints/step_300/config.json

@@ -0,0 +1,4 @@
+{
+  "model_type": "bert",
+  "architectures": ["BertModel"]
+}

checkpoints/step_300/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:965362299a238de576a92dfdd3e32aea7a2bacc94b2c41541c8c9258b923f587
+size 23

checkpoints/step_400/config.json

@@ -0,0 +1,4 @@
+{
+    "model_type": "bert",
+    "architectures": ["BertModel"]
+  }

checkpoints/step_400/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:965362299a238de576a92dfdd3e32aea7a2bacc94b2c41541c8c9258b923f587
+size 23

checkpoints/step_500/config.json

@@ -0,0 +1,4 @@
+{
+    "model_type": "bert",
+    "architectures": ["BertModel"]
+  }

checkpoints/step_500/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:965362299a238de576a92dfdd3e32aea7a2bacc94b2c41541c8c9258b923f587
+size 23

checkpoints/step_600/config.json

@@ -0,0 +1,4 @@
+{
+    "model_type": "bert",
+    "architectures": ["BertModel"]
+  }

checkpoints/step_600/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:965362299a238de576a92dfdd3e32aea7a2bacc94b2c41541c8c9258b923f587
+size 23

checkpoints/step_700/config.json

@@ -0,0 +1,4 @@
+{
+    "model_type": "bert",
+    "architectures": ["BertModel"]
+  }

checkpoints/step_700/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:965362299a238de576a92dfdd3e32aea7a2bacc94b2c41541c8c9258b923f587
+size 23

checkpoints/step_800/config.json

@@ -0,0 +1,4 @@
+{
+    "model_type": "bert",
+    "architectures": ["BertModel"]
+  }

checkpoints/step_800/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:965362299a238de576a92dfdd3e32aea7a2bacc94b2c41541c8c9258b923f587
+size 23

checkpoints/step_900/config.json

@@ -0,0 +1,4 @@
+{
+    "model_type": "bert",
+    "architectures": ["BertModel"]
+  }

checkpoints/step_900/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:965362299a238de576a92dfdd3e32aea7a2bacc94b2c41541c8c9258b923f587
+size 23

evaluation/benchmarks/code_generation/eval.py

@@ -0,0 +1,33 @@
+import argparse
+import os
+import sys
+
+# Add parent directory to path to import utils
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..'))
+from utils.benchmark_utils import get_benchmark_score
+
+def main():
+    parser = argparse.ArgumentParser(description="Evaluate code_generation")
+    parser.add_argument("model_path", type=str, help="Path to model checkpoint")
+    args = parser.parse_args()
+    
+    if not os.path.isfile(args.model_path):
+        print(f"Error: Directory not found at '{args.model_path}'", file=sys.stderr)
+        sys.exit(1)
+    
+    checkpoint_name = os.path.basename(os.path.normpath(args.model_path))
+    try:
+        step_number = int(checkpoint_name.split('_')[-1])
+    except (ValueError, IndexError):
+        print(f"Error: Cannot parse step number from '{checkpoint_name}'", file=sys.stderr)
+        sys.exit(1)
+    
+    result = get_benchmark_score("code_generation", step_number)
+    if result is None:
+        print(f"Error: Invalid step number {step_number}", file=sys.stderr)
+        sys.exit(1)
+    
+    print(result)
+
+if __name__ == "__main__":
+    main()

evaluation/benchmarks/common_sense/eval.py

@@ -0,0 +1,33 @@
+import argparse
+import os
+import sys
+
+# Add parent directory to path to import utils
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..'))
+from utils.benchmark_utils import get_benchmark_score
+
+def main():
+    parser = argparse.ArgumentParser(description="Evaluate common_sense")
+    parser.add_argument("model_path", type=str, help="Path to model checkpoint")
+    args = parser.parse_args()
+    
+    if not os.path.isdir(args.model_path):
+        print(f"Error: Directory not found at '{args.model_path}'", file=sys.stderr)
+        sys.exit(1)
+    
+    checkpoint_name = os.path.basename(os.path.normpath(args.model_path))
+    try:
+        step_number = int(checkpoint_name.split('_')[-1])
+    except (ValueError, IndexError):
+        print(f"Error: Cannot parse step number from '{checkpoint_name}'", file=sys.stderr)
+        sys.exit(1)
+    
+    result = get_benchmark_score("common_sense", step_number)
+    if result is None:
+        print(f"Error: Invalid step number {step_number}", file=sys.stderr)
+        sys.exit(1)
+    
+    print(result)
+
+if __name__ == "__main__":
+    main()

evaluation/benchmarks/creative_writing/eval.py

@@ -0,0 +1,33 @@
+import argparse
+import os
+import sys
+
+# Add parent directory to path to import utils
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..'))
+from utils.benchmark_utils import get_benchmark_score
+
+def main():
+    parser = argparse.ArgumentParser(description="Evaluate creative_writing")
+    parser.add_argument("model_path", type=str, help="Path to model checkpoint")
+    args = parser.parse_args()
+    
+    if not os.path.isdir(args.model_path):
+        print(f"Error: Directory not found at '{args.model_path}'", file=sys.stderr)
+        sys.exit(1)
+    
+    checkpoint_name = os.path.basename(os.path.normpath(args.model_path))
+    try:
+        step_number = int(checkpoint_name.split('_')[-1])
+    except (ValueError, IndexError):
+        print(f"Error: Cannot parse step number from '{checkpoint_name}'", file=sys.stderr)
+        sys.exit(1)
+    
+    result = get_benchmark_score("creative_writing", step_number)
+    if result is None:
+        print(f"Error: Invalid step number {step_number}", file=sys.stderr)
+        sys.exit(1)
+    
+    print(result)
+
+if __name__ == "__main__":
+    main()

evaluation/benchmarks/dialogue_generation/eval.py

@@ -0,0 +1,35 @@
+import argparse
+import os
+import sys
+
+# Add parent directory to path to import utils
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..'))
+from utils.benchmark_utils import get_benchmark_score
+
+def main():
+    parser = argparse.ArgumentParser(description="Evaluate dialogue generation")
+    parser.add_argument("model_path", type=str, help="Path to model checkpoint")
+    args = parser.parse_args()
+    
+    if not os.path.isdir(args.model_path):
+        print(f"Error: Directory not found at '{args.model_path}'", file=sys.stderr)
+        sys.exit(1)
+    
+    checkpoint_name = os.path.basename(os.path.normpath(args.model_path))
+    try:
+        step_number = int(checkpoint_name.split('_')[-1])
+    except (ValueError, IndexError):
+        print(f"Error: Cannot parse step number from '{checkpoint_name}'", file=sys.stderr)
+        sys.exit(1)
+    
+    config_data = config_init()
+    
+    result = get_benchmark_score("dialogue_generation", step_number)
+    if result is None:
+        print(f"Error: Invalid step number {step_number}", file=sys.stderr)
+        sys.exit(1)
+    
+    print(result)
+
+if __name__ == "__main__":
+    main()

evaluation/benchmarks/instruction_following/eval.py

@@ -0,0 +1,33 @@
+import argparse
+import os
+import sys
+
+# Add parent directory to path to import utils
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..'))
+from utils.benchmark_utils import get_benchmark_score
+
+def main():
+    parser = argparse.ArgumentParser(description="Evaluate instruction_following")
+    parser.add_argument("model_path", type=str, help="Path to model checkpoint")
+    args = parser.parse_args()
+    
+    if not os.path.isdir(args.model_path):
+        print(f"Error: Directory not found at '{args.model_path}'", file=sys.stderr)
+        sys.exit(1)
+    
+    checkpoint_name = os.path.basename(os.path.normpath(args.model_path))
+    try:
+        step_number = int(checkpoint_name.split('_')[-1])
+    except (ValueError, IndexError):
+        print(f"Error: Cannot parse step number from '{checkpoint_name}'", file=sys.stderr)
+        sys.exit(1)
+    
+    result = get_benchmark_score("instruction_following", step_number)
+    if result is None:
+        print(f"Error: Invalid step number {step_number}", file=sys.stderr)
+        sys.exit(1)
+    
+    print(result)
+
+if __name__ == "__main__":
+    main()

evaluation/benchmarks/knowledge_retrieval/eval.py

@@ -0,0 +1,33 @@
+import argparse
+import os
+import sys
+
+# Add parent directory to path to import utils
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..'))
+from utils.benchmark_utils import get_benchmark_score
+
+def main():
+    parser = argparse.ArgumentParser(description="Evaluate knowledge_retrieval")
+    parser.add_argument("model_path", type=str, help="Path to model checkpoint")
+    args = parser.parse_args()
+    
+    if not os.path.isdir(args.model_path):
+        print(f"Error: Directory not found at '{args.model_path}'", file=sys.stderr)
+        sys.exit(1)
+    
+    checkpoint_name = os.path.basename(os.path.normpath(args.model_path))
+    try:
+        step_number = int(checkpoint_name.split('_')[-1])
+    except (ValueError, IndexError):
+        print(f"Error: Cannot parse step number from '{checkpoint_name}'", file=sys.stderr)
+        sys.exit(1)
+    
+    result = get_benchmark_score("knowledge_retrieval", step_number)
+    if result is None:
+        print(f"Error: Invalid step number {step_number}", file=sys.stderr)
+        sys.exit(1)
+    
+    print(result)
+
+if __name__ == "__main__":
+    main()

evaluation/benchmarks/logical_reasoning/eval.py

@@ -0,0 +1,33 @@
+import argparse
+import os
+import sys
+
+# Add parent directory to path to import utils
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..'))
+from utils.benchmark_utils import get_benchmark_score
+
+def main():
+    parser = argparse.ArgumentParser(description="Evaluate logical_reasoning")
+    parser.add_argument("model_path", type=str, help="Path to model checkpoint")
+    args = parser.parse_args()
+    
+    if not os.path.isdir(args.model_path):
+        print(f"Error: Directory not found at '{args.model_path}'", file=sys.stderr)
+        sys.exit(1)
+    
+    checkpoint_name = os.path.basename(os.path.normpath(args.model_path))
+    try:
+        step_number = int(checkpoint_name.split('_')[-1])
+    except (ValueError, IndexError):
+        print(f"Error: Cannot parse step number from '{checkpoint_name}'", file=sys.stderr)
+        sys.exit(1)
+    
+    result = get_benchmark_score("logical_reasoning", step_number)
+    if result is None:
+        print(f"Error: Invalid step number {step_number}", file=sys.stderr)
+        sys.exit(1)
+    
+    print(result)
+
+if __name__ == "__main__":
+    main()

evaluation/benchmarks/math_reasoning/eval.py

@@ -0,0 +1,33 @@
+import argparse
+import os
+import sys
+
+# Add parent directory to path to import utils
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..'))
+from utils.benchmark_utils import get_benchmark_score
+
+def main():
+    parser = argparse.ArgumentParser(description="Evaluate math_reasoning")
+    parser.add_argument("model_path", type=str, help="Path to model checkpoint")
+    args = parser.parse_args()
+    
+    if not os.path.isdir(args.model_path):
+        print(f"Error: Directory not found at '{args.model_path}'", file=sys.stderr)
+        sys.exit(1)
+    
+    checkpoint_name = os.path.basename(os.path.normpath(args.model_path))
+    try:
+        step_number = int(checkpoint_name.split('_')[-1])
+    except (ValueError, IndexError):
+        print(f"Error: Cannot parse step number from '{checkpoint_name}'", file=sys.stderr)
+        sys.exit(1)
+    
+    result = get_benchmark_score("math_reasoning", step_number)
+    if result is None:
+        print(f"Error: Invalid step number {step_number}", file=sys.stderr)
+        sys.exit(1)
+    
+    print(result)
+
+if __name__ == "__main__":
+    main()

evaluation/benchmarks/question_answering/eval.py

@@ -0,0 +1,33 @@
+import argparse
+import os
+import sys
+
+# Add parent directory to path to import utils
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..'))
+from utils.benchmark_utils import get_benchmark_score
+
+def main():
+    parser = argparse.ArgumentParser(description="Evaluate question_answering")
+    parser.add_argument("model_path", type=str, help="Path to model checkpoint")
+    args = parser.parse_args()
+    
+    if not os.path.isdir(args.model_path):
+        print(f"Error: Directory not found at '{args.model_path}'", file=sys.stderr)
+        sys.exit(1)
+    
+    checkpoint_name = os.path.basename(os.path.normpath(args.model_path))
+    try:
+        step_number = int(checkpoint_name.split('_')[-1])
+    except (ValueError, IndexError):
+        print(f"Error: Cannot parse step number from '{checkpoint_name}'", file=sys.stderr)
+        sys.exit(1)
+    
+    result = get_benchmark_score("question_answering", step_number)
+    if result is None:
+        print(f"Error: Invalid step number {step_number}", file=sys.stderr)
+        sys.exit(1)
+    
+    print(result)
+
+if __name__ == "__main__":
+    main()

evaluation/benchmarks/reading_comprehension/eval.py

@@ -0,0 +1,33 @@
+import argparse
+import os
+import sys
+
+# Add parent directory to path to import utils
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..'))
+from utils.benchmark_utils import get_benchmark_score
+
+def main():
+    parser = argparse.ArgumentParser(description="Evaluate reading_comprehension")
+    parser.add_argument("model_path", type=str, help="Path to model checkpoint")
+    args = parser.parse_args()
+    
+    if not os.path.isdir(args.model_path):
+        print(f"Error: Directory not found at '{args.model_path}'", file=sys.stderr)
+        sys.exit(1)
+    
+    checkpoint_name = os.path.basename(os.path.normpath(args.model_path))
+    try:
+        step_number = int(checkpoint_name.split('_')[-1])
+    except (ValueError, IndexError):
+        print(f"Error: Cannot parse step number from '{checkpoint_name}'", file=sys.stderr)
+        sys.exit(1)
+    
+    result = get_benchmark_score("reading_comprehension", step_number)
+    if result is None:
+        print(f"Error: Invalid step number {step_number}", file=sys.stderr)
+        sys.exit(1)
+    
+    print(result)
+
+if __name__ == "__main__":
+    main()

evaluation/benchmarks/safety_evaluation/eval.py

@@ -0,0 +1,33 @@
+import argparse
+import os
+import sys
+
+# Add parent directory to path to import utils
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..'))
+from utils.benchmark_utils import get_benchmark_score
+
+def main():
+    parser = argparse.ArgumentParser(description="Evaluate safety_evaluation")
+    parser.add_argument("model_path", type=str, help="Path to model checkpoint")
+    args = parser.parse_args()
+    
+    if not os.path.isdir(args.model_path):
+        print(f"Error: Directory not found at '{args.model_path}'", file=sys.stderr)
+        sys.exit(1)
+    
+    checkpoint_name = os.path.basename(os.path.normpath(args.model_path))
+    try:
+        step_number = int(checkpoint_name.split('_')[-1])
+    except (ValueError, IndexError):
+        print(f"Error: Cannot parse step number from '{checkpoint_name}'", file=sys.stderr)
+        sys.exit(1)
+    
+    result = get_benchmark_score("safety_evaluation", step_number)
+    if result is None:
+        print(f"Error: Invalid step number {step_number}", file=sys.stderr)
+        sys.exit(1)
+    
+    print(result)
+
+if __name__ == "__main__":
+    main()

evaluation/benchmarks/sentiment_analysis/eval.py

@@ -0,0 +1,33 @@
+import argparse
+import os
+import sys
+
+# Add parent directory to path to import utils
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..'))
+from utils.benchmark_utils import get_benchmark_score
+
+def main():
+    parser = argparse.ArgumentParser(description="Evaluate sentiment_analysis")
+    parser.add_argument("model_path", type=str, help="Path to model checkpoint")
+    args = parser.parse_args()
+    
+    if not os.path.isdir(args.model_path):
+        print(f"Error: Directory not found at '{args.model_path}'", file=sys.stderr)
+        sys.exit(1)
+    
+    checkpoint_name = os.path.basename(os.path.normpath(args.model_path))
+    try:
+        step_number = int(checkpoint_name.split('_')[-1])
+    except (ValueError, IndexError):
+        print(f"Error: Cannot parse step number from '{checkpoint_name}'", file=sys.stderr)
+        sys.exit(1)
+    
+    result = get_benchmark_score("sentiment_analysis", step_number)
+    if result is None:
+        print(f"Error: Invalid step number {step_number}", file=sys.stderr)
+        sys.exit(1)
+    
+    print(result)
+
+if __name__ == "__main__":
+    main()

evaluation/benchmarks/summarization/eval.py

@@ -0,0 +1,33 @@
+import argparse
+import os
+import sys
+
+# Add parent directory to path to import utils
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..'))
+from utils.benchmark_utils import get_benchmark_score
+
+def main():
+    parser = argparse.ArgumentParser(description="Evaluate summarization")
+    parser.add_argument("model_path", type=str, help="Path to model checkpoint")
+    args = parser.parse_args()
+    
+    if not os.path.isdir(args.model_path):
+        print(f"Error: Directory not found at '{args.model_path}'", file=sys.stderr)
+        sys.exit(1)
+    
+    checkpoint_name = os.path.basename(os.path.normpath(args.model_path))
+    try:
+        step_number = int(checkpoint_name.split('_')[-1])
+    except (ValueError, IndexError):
+        print(f"Error: Cannot parse step number from '{checkpoint_name}'", file=sys.stderr)
+        sys.exit(1)
+    
+    result = get_benchmark_score("summarization", step_number)
+    if result is None:
+        print(f"Error: Invalid step number {step_number}", file=sys.stderr)
+        sys.exit(1)
+    
+    print(result)
+
+if __name__ == "__main__":
+    main()

evaluation/benchmarks/text_classification/eval.py

@@ -0,0 +1,34 @@
+import argparse
+import os
+import sys
+import util
+
+# Add parent directory to path to import utils
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..'))
+from utils.benchmark_utils import get_benchmark_score
+
+def main():
+    parser = argparse.ArgumentParser(description="Evaluate text classification")
+    parser.add_argument("model_path", type=str, help="Path to model checkpoint")
+    args = parser.parse_args()
+    
+    if not os.path.isdir(args.model_path):
+        print(f"Error: Directory not found at '{args.model_path}'", file=sys.stderr)
+        sys.exit(1)
+    
+    checkpoint_name = os.path.basename(os.path.normpath(args.model_path))
+    try:
+        step_number = int(checkpoint_name.split('_')[-1])
+    except (ValueError, IndexError):
+        print(f"Error: Cannot parse step number from '{checkpoint_name}'", file=sys.stderr)
+        sys.exit(1)
+    
+    result = get_benchmark_score("text_classification", step_number)
+    if result is None:
+        print(f"Error: Invalid step number {step_number}", file=sys.stderr)
+        sys.exit(1)
+    
+    print(result)
+
+if __name__ == "__main__":
+    main()

evaluation/benchmarks/translation/eval.py

@@ -0,0 +1,33 @@
+import argparse
+import os
+import sys
+
+# Add parent directory to path to import utils
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..'))
+from utils.benchmark_utils import get_benchmark_score
+
+def main():
+    parser = argparse.ArgumentParser(description="Evaluate translation")
+    parser.add_argument("model_path", type=str, help="Path to model checkpoint")
+    args = parser.parse_args()
+    
+    if not os.path.isdir(args.model_path):
+        print(f"Error: Directory not found at '{args.model_path}'", file=sys.stderr)
+        sys.exit(1)
+    
+    checkpoint_name = os.path.basename(os.path.normpath(args.model_path))
+    try:
+        step_number = int(checkpoint_name.split('_')[-1])
+    except (ValueError, IndexError):
+        print(f"Error: Cannot parse step number from '{checkpoint_name}'", file=sys.stderr)
+        sys.exit(1)
+    
+    result = get_benchmark_score("translation", step_number)
+    if result is None:
+        print(f"Error: Invalid step number {step_number}", file=sys.stderr)
+        sys.exit(1)
+    
+    print(result)
+
+if __name__ == "__main__":
+    main()

evaluation/build/lib.linux-x86_64-cpython-313/utils/benchmark_utils.cpython-313-x86_64-linux-gnu.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9ae9c7cc713b5dae1e04fa9c128874564d866648bed5e7f465adf34785d0d212
+size 713688

evaluation/build/temp.linux-x86_64-cpython-313/utils/benchmark_utils.o

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:357ac47110898a21c3619d820c20f5841e7c019f98099be33b615709100ecb21
+size 1385208

evaluation/eval.py

@@ -0,0 +1,122 @@
+import argparse
+import os
+import sys
+import subprocess
+
+# Add utils to path
+sys.path.insert(0, os.path.dirname(__file__))
+from utils.benchmark_utils import BENCHMARK_CALCULATORS
+
+# List of all benchmark categories
+BENCHMARK_CATEGORIES = list(BENCHMARK_CALCULATORS.keys())
+
+def run_benchmark_evaluation(benchmark_name, model_path):
+    """Run evaluation for a specific benchmark category"""
+    benchmark_script = os.path.join("evaluation", "benchmarks", benchmark_name, "eval.py")
+    
+    if not os.path.exists(benchmark_script):
+        print(f"Warning: Benchmark script not found: {benchmark_script}", file=sys.stderr)
+        return None
+    
+    try:
+        result = subprocess.run(
+            [sys.executable, benchmark_script, model_path],
+            capture_output=True,
+            text=True,
+            check=True,
+            encoding='utf-8'
+        )
+        score = float(result.stdout.strip())
+        return score
+    except subprocess.CalledProcessError as e:
+        print(f"Error running {benchmark_name} evaluation: {e.stderr}", file=sys.stderr)
+        return None
+    except (ValueError, TypeError):
+        print(f"Warning: Could not parse score from {benchmark_name}: '{result.stdout.strip()}'", file=sys.stderr)
+        return None
+
+def calculate_overall_score(benchmark_scores):
+    """Calculate overall performance score from individual benchmarks"""
+    valid_scores = [score for score in benchmark_scores.values() if score is not None]
+    if not valid_scores:
+        return None
+    
+    # Weighted average with slight emphasis on reasoning tasks
+    weights = {
+        "math_reasoning": 1.2,
+        "logical_reasoning": 1.2, 
+        "code_generation": 1.1,
+        "question_answering": 1.1,
+        "reading_comprehension": 1.0,
+        "common_sense": 1.0,
+        "text_classification": 0.9,
+        "sentiment_analysis": 0.9,
+        "dialogue_generation": 1.0,
+        "summarization": 1.0,
+        "translation": 1.0,
+        "knowledge_retrieval": 1.0,
+        "creative_writing": 0.9,
+        "instruction_following": 1.1,
+        "safety_evaluation": 1.1
+    }
+    
+    weighted_sum = 0
+    total_weight = 0
+    
+    for benchmark, score in benchmark_scores.items():
+        if score is not None:
+            weight = weights.get(benchmark, 1.0)
+            weighted_sum += score * weight
+            total_weight += weight
+    
+    return round(weighted_sum / total_weight, 3) if total_weight > 0 else None
+
+def main():
+    """
+    Run comprehensive evaluation across all benchmark categories.
+    Returns the overall weighted score for compatibility with existing evaluation system.
+    """
+    parser = argparse.ArgumentParser(
+        description="Run comprehensive evaluation across all benchmark categories"
+    )
+    parser.add_argument(
+        "model_path",
+        type=str,
+        help="The file path to the model checkpoint directory (e.g., ../checkpoints/step_100)."
+    )
+    args = parser.parse_args()
+
+    # Check if the provided path is a directory
+    if not os.path.isdir(args.model_path):
+        print(f"Error: Directory not found at '{args.model_path}'", file=sys.stderr)
+        sys.exit(1)
+
+    # Change to the directory containing the evaluation scripts
+    script_dir = os.path.dirname(os.path.abspath(__file__))
+    original_cwd = os.getcwd()
+    os.chdir(os.path.dirname(script_dir))
+
+    benchmark_scores = {}
+    
+    # Run evaluation for each benchmark category
+    for benchmark in BENCHMARK_CATEGORIES:
+        score = run_benchmark_evaluation(benchmark, args.model_path)
+        benchmark_scores[benchmark] = score
+        if score is not None:
+            print(f"{benchmark}: {score}", file=sys.stderr)
+    
+    # Calculate overall score
+    overall_score = calculate_overall_score(benchmark_scores)
+    
+    # Restore original working directory
+    os.chdir(original_cwd)
+    
+    if overall_score is None:
+        print(f"Error: Could not calculate overall score for {args.model_path}", file=sys.stderr)
+        sys.exit(1)
+
+    # Print only the overall score for compatibility with existing evaluation pipeline
+    print(overall_score)
+
+if __name__ == "__main__":
+    main()

evaluation/setup.py

@@ -0,0 +1,19 @@
+from setuptools import setup, Extension
+from Cython.Build import cythonize
+
+# Let Cython find and compile your .py files directly.
+# This compiles both __init__.py and benchmark_utils.py
+
+extensions = [
+    Extension("utils.__init__", ["utils/__init__.py"]),
+    Extension("utils.benchmark_utils", ["utils/benchmark_utils.py"]),
+]
+
+setup(
+    name="my_utils_package",
+    ext_modules=cythonize(
+        extensions,
+        # Tell Cython you're using Python 3 syntax
+        compiler_directives={'language_level' : "3"}
+    )
+)

evaluation/utils/benchmark_utils.cpython-313-x86_64-linux-gnu.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9ae9c7cc713b5dae1e04fa9c128874564d866648bed5e7f465adf34785d0d212
+size 713688