File tree Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Original file line number Diff line number Diff line change 1
1
# Copyright (c) Meta Platforms, Inc. and affiliates.
2
2
# All rights reserved.
3
- # Copyright 2024 Arm Limited and/or its affiliates.
3
+ # Copyright 2024-2025 Arm Limited and/or its affiliates.
4
4
#
5
5
# This source code is licensed under the BSD-style license found in the
6
6
# LICENSE file in the root directory of this source tree.
@@ -37,6 +37,7 @@ class Model(str, Enum):
37
37
EfficientSam = "efficient_sam"
38
38
Qwen25 = "qwen2_5"
39
39
Phi4Mini = "phi_4_mini"
40
+ SmolLM2 = "smollm2"
40
41
41
42
def __str__ (self ) -> str :
42
43
return self .value
@@ -82,6 +83,7 @@ def __str__(self) -> str:
82
83
str (Model .EfficientSam ): ("efficient_sam" , "EfficientSAM" ),
83
84
str (Model .Qwen25 ): ("qwen2_5" , "Qwen2_5Model" ),
84
85
str (Model .Phi4Mini ): ("phi_4_mini" , "Phi4MiniModel" ),
86
+ str (Model .SmolLM2 ): ("smollm2" , "SmolLM2Model" ),
85
87
}
86
88
87
89
__all__ = [
You can’t perform that action at this time.
0 commit comments