Skip to content

Commit 2f47ee2

Browse files
authored
Adding smollm2 to examples/models/__init__.py (#13514)
### Summary Adding SmolLM2 to list of models for use in testing.
1 parent 1356118 commit 2f47ee2

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

examples/models/__init__.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
# Copyright (c) Meta Platforms, Inc. and affiliates.
22
# All rights reserved.
3-
# Copyright 2024 Arm Limited and/or its affiliates.
3+
# Copyright 2024-2025 Arm Limited and/or its affiliates.
44
#
55
# This source code is licensed under the BSD-style license found in the
66
# LICENSE file in the root directory of this source tree.
@@ -37,6 +37,7 @@ class Model(str, Enum):
3737
EfficientSam = "efficient_sam"
3838
Qwen25 = "qwen2_5"
3939
Phi4Mini = "phi_4_mini"
40+
SmolLM2 = "smollm2"
4041

4142
def __str__(self) -> str:
4243
return self.value
@@ -82,6 +83,7 @@ def __str__(self) -> str:
8283
str(Model.EfficientSam): ("efficient_sam", "EfficientSAM"),
8384
str(Model.Qwen25): ("qwen2_5", "Qwen2_5Model"),
8485
str(Model.Phi4Mini): ("phi_4_mini", "Phi4MiniModel"),
86+
str(Model.SmolLM2): ("smollm2", "SmolLM2Model"),
8587
}
8688

8789
__all__ = [

0 commit comments

Comments
 (0)