Skip to content

Commit 14d3ca7

Browse files
author
Danqing Wang (MPK)
committed
update model name to smollm2
1 parent 91e5dd2 commit 14d3ca7

File tree

5 files changed

+3
-3
lines changed

5 files changed

+3
-3
lines changed

examples/models/llama/export_llama_lib.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,7 @@
9494
"static_llama",
9595
"qwen2_5",
9696
"phi-4-mini",
97-
"smolllm2",
97+
"smollm2",
9898
]
9999
TORCHTUNE_DEFINED_MODELS = ["llama3_2_vision"]
100100

examples/models/smollm/README.md

Whitespace-only changes.

examples/models/smollm/__init__ renamed to examples/models/smollm2/__init__

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,11 +4,11 @@
44
from executorch.example.models.llama.model import Llama2Model
55

66

7-
class SmolLMModel(Llama2Model):
7+
class SmolLM2Model(Llama2Model):
88
def __init__(self, **kwargs):
99
super().__init__(**kwargs)
1010

1111

1212
__all__ = [
13-
"SmolLMModel",
13+
"SmolLM2Model",
1414
]

0 commit comments

Comments
 (0)