From 5b3e7f1fa7c3b00979c69240713aad1119829795 Mon Sep 17 00:00:00 2001
From: Shane O'Brien <shobrien@amd.com>
Date: Thu, 12 Dec 2024 10:45:43 +0000
Subject: [PATCH] Update readmes and add OLMo and Qwen to the CI tests_utils

---
 README.md                      | 2 +-
 src/python/py/models/README.md | 1 +
 test/python/_test_utils.py     | 2 ++
 3 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 8da6db37e..6503c8f63 100644
--- a/README.md
+++ b/README.md
@@ -15,7 +15,7 @@ See documentation at https://onnxruntime.ai/docs/genai.
 
 | Support matrix | Supported now | Under development | On the roadmap |
 | -------------- | ------------- | ----------------- | -------------- |
-| Model architectures | Gemma <br/> Llama * <br/> Mistral + <br/> Phi (language + vision) <br/> Qwen <br/> Nemotron <br/> Granite <br/> | Whisper | Stable diffusion |
+| Model architectures | Gemma <br/> Llama * <br/> Mistral + <br/> Phi (language + vision) <br/> Qwen <br/> Nemotron <br/> Granite <br/> AMD OLMo | Whisper | Stable diffusion |
 | API | Python <br/> C# <br/> C/C++ <br/> Java ^ | Objective-C | |
 | Platform | Linux <br/> Windows <br/> Mac ^ <br/> Android ^ | | iOS |
 | Architecture | x86 <br/> x64 <br/> Arm64 ~ | | |
diff --git a/src/python/py/models/README.md b/src/python/py/models/README.md
index e216a0ca1..c10683ef1 100644
--- a/src/python/py/models/README.md
+++ b/src/python/py/models/README.md
@@ -39,6 +39,7 @@ The tool currently supports the following model architectures.
 - Nemotron
 - Phi
 - Qwen
+- AMD OLMo
 
 It is intended for supporting the latest, popular state-of-the-art models.
 
diff --git a/test/python/_test_utils.py b/test/python/_test_utils.py
index da5785bb3..d5a2aef10 100644
--- a/test/python/_test_utils.py
+++ b/test/python/_test_utils.py
@@ -55,6 +55,8 @@ def run_subprocess(
 def get_model_paths():
     hf_paths = {
         "phi-2": "microsoft/phi-2",
+        "olmo": "amd/AMD-OLMo-1B-SFT-DPO",
+        "qwen": "Qwen/Qwen2.5-0.5B",
         "phi-3.5": "microsoft/Phi-3.5-mini-instruct",
         # "llama-3.2": "meta-llama/Llama-3.2-1B-instruct",
         "granite-3.0": "ibm-granite/granite-3.0-2b-instruct",