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 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__ = [
You can’t perform that action at this time.
0 commit comments