Arm backend: Added support for example models.

Added missing enablement of model definitions within examples/models
for model testing.

Change-Id: I4fc8cb7377821b1ba2c262da692e016713418f70
Signed-off-by: Michiel Olieslagers <michiel.olieslagers@arm.com>

Author: Michiel-Olieslagers

examples/models/BUCK

@@ -33,6 +33,9 @@ fbcode_target(_kind = python_library,
         "//executorch/examples/models/phi_4_mini:phi_4_mini",  # @manual
         "//executorch/examples/models/smollm2:smollm2",  # @manual
         "//executorch/examples/models/smollm3:smollm3",  # @manual
+        "//executorch/examples/models/smolvlm:smolvlm",  # @manual
+        "//executorch/examples/models/whisper:whisper",  # @manual
+        "//executorch/examples/models/yolo26:yolo26",  # @manual
     ],
 )
 

examples/models/__init__.py

@@ -45,6 +45,10 @@ class Model(str, Enum):
     MobileNetV1025 = "mobilenet_v1_025"
     ResNet8 = "resnet8"
     Sdpa = "sdpa"
+    Qwen3 = "qwen3"
+    SmolVLM = "smolvlm"
+    YOLO26 = "yolo26"
+    Whisper = "whisper"
 
     def __str__(self) -> str:
         return self.value
@@ -105,6 +109,10 @@ def __str__(self) -> str:
     ),
     str(Model.ResNet8): ("mlperf_tiny.resnet8", "ResNet8Model"),
     str(Model.Sdpa): ("toy_model", "SdpaModule"),
+    str(Model.Qwen3): ("qwen3", "Qwen3Model"),
+    str(Model.SmolVLM): ("smolvlm", "SmolVLMModel"),
+    str(Model.YOLO26): ("yolo26", "YOLO26Model"),
+    str(Model.Whisper): ("whisper", "WhisperModel"),
 }
 
 __all__ = [

examples/models/whisper/__init__.py

@@ -0,0 +1,8 @@
+# Copyright 2026 Arm Limited and/or its affiliates.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+from .model import WhisperModel
+
+__all__ = [WhisperModel]

examples/models/whisper/model.py

@@ -0,0 +1,43 @@
+# Copyright 2026 Arm Limited and/or its affiliates.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+import logging
+
+import torch
+from transformers import WhisperModel as wm
+
+from ..model_base import EagerModelBase
+
+
+class WhisperWrapper(torch.nn.Module):
+    def __init__(self, model):
+        super().__init__()
+        self.model = model
+
+    def forward(self, input_features, decoder_input_ids):
+        return self.model(
+            input_features=input_features,
+            decoder_input_ids=decoder_input_ids,
+            use_cache=False,
+        ).last_hidden_state
+
+
+class WhisperModel(EagerModelBase):
+    def __init__(self, model_name: str = "openai/whisper-base"):
+        self.model = wm.from_pretrained(model_name)
+
+    def get_eager_model(self) -> torch.nn.Module:
+        logging.info("Loading Whisper model")
+        m = WhisperWrapper(self.model).eval()
+        logging.info("Loaded Whisper model")
+        return m
+
+    def get_example_inputs(self):
+        return (
+            torch.randn(1, 80, 3000),
+            torch.tensor(
+                [[self.model.config.decoder_start_token_id]], dtype=torch.long
+            ),
+        )

examples/models/yolo26/__init__.py

@@ -0,0 +1,8 @@
+# Copyright 2026 Arm Limited and/or its affiliates.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+from .model import YOLO26Model
+
+__all__ = [YOLO26Model]

examples/models/yolo26/model.py

@@ -0,0 +1,30 @@
+# Copyright 2026 Arm Limited and/or its affiliates.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+import logging
+
+import torch
+
+from ultralytics import YOLO
+
+from ..model_base import EagerModelBase
+
+
+class YOLO26Model(EagerModelBase):
+    def __init__(self, model_name: str = "yolo26n.pt"):
+        self.model_name = model_name
+        self.yolo = YOLO(model_name)
+        self.dummy_frame = torch.randn((320, 320, 3)).to(torch.uint8).numpy()
+        self.yolo.predict(self.dummy_frame, imgsz=(320, 320), verbose=False)
+        self.model = self.yolo.model.eval()
+
+    def get_eager_model(self) -> torch.nn.Module:
+        logging.info("Loading " + self.model_name + " model")
+        m = self.model
+        logging.info("Loaded " + self.model_name + " model")
+        return m
+
+    def get_example_inputs(self):
+        return (self.yolo.predictor.preprocess([self.dummy_frame]),)