Skip to content

Commit 5b3e7f1

Browse files
committed
Update readmes and add OLMo and Qwen to the CI tests_utils
1 parent d0495f2 commit 5b3e7f1

File tree

3 files changed

+4
-1
lines changed

3 files changed

+4
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ See documentation at https://onnxruntime.ai/docs/genai.
1515

1616
| Support matrix | Supported now | Under development | On the roadmap |
1717
| -------------- | ------------- | ----------------- | -------------- |
18-
| Model architectures | Gemma <br/> Llama * <br/> Mistral + <br/> Phi (language + vision) <br/> Qwen <br/> Nemotron <br/> Granite <br/> | Whisper | Stable diffusion |
18+
| Model architectures | Gemma <br/> Llama * <br/> Mistral + <br/> Phi (language + vision) <br/> Qwen <br/> Nemotron <br/> Granite <br/> AMD OLMo | Whisper | Stable diffusion |
1919
| API | Python <br/> C# <br/> C/C++ <br/> Java ^ | Objective-C | |
2020
| Platform | Linux <br/> Windows <br/> Mac ^ <br/> Android ^ | | iOS |
2121
| Architecture | x86 <br/> x64 <br/> Arm64 ~ | | |

src/python/py/models/README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@ The tool currently supports the following model architectures.
3939
- Nemotron
4040
- Phi
4141
- Qwen
42+
- AMD OLMo
4243

4344
It is intended for supporting the latest, popular state-of-the-art models.
4445

test/python/_test_utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,8 @@ def run_subprocess(
5555
def get_model_paths():
5656
hf_paths = {
5757
"phi-2": "microsoft/phi-2",
58+
"olmo": "amd/AMD-OLMo-1B-SFT-DPO",
59+
"qwen": "Qwen/Qwen2.5-0.5B",
5860
"phi-3.5": "microsoft/Phi-3.5-mini-instruct",
5961
# "llama-3.2": "meta-llama/Llama-3.2-1B-instruct",
6062
"granite-3.0": "ibm-granite/granite-3.0-2b-instruct",

0 commit comments

Comments
 (0)