Skip to content

Commit b617f95

Browse files
committed
Rename SmolLM3 to SmolLm3 for consistency
1 parent 9d57fac commit b617f95

4 files changed

Lines changed: 13 additions & 13 deletions

File tree

lib/bumblebee.ex

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -211,11 +211,11 @@ defmodule Bumblebee do
211211
"RobertaForTokenClassification" => {Bumblebee.Text.Roberta, :for_token_classification},
212212
"RobertaForCausalLM" => {Bumblebee.Text.Roberta, :for_causal_language_modeling},
213213
"RobertaModel" => {Bumblebee.Text.Roberta, :base},
214-
"SmolLM3Model" => {Bumblebee.Text.SmolLM3, :base},
215-
"SmolLM3ForCausalLM" => {Bumblebee.Text.SmolLM3, :for_causal_language_modeling},
216-
"SmolLM3ForQuestionAnswering" => {Bumblebee.Text.SmolLM3, :for_question_answering},
217-
"SmolLM3ForSequenceClassification" => {Bumblebee.Text.SmolLM3, :for_sequence_classification},
218-
"SmolLM3ForTokenClassification" => {Bumblebee.Text.SmolLM3, :for_token_classification},
214+
"SmolLM3Model" => {Bumblebee.Text.SmolLm3, :base},
215+
"SmolLM3ForCausalLM" => {Bumblebee.Text.SmolLm3, :for_causal_language_modeling},
216+
"SmolLM3ForQuestionAnswering" => {Bumblebee.Text.SmolLm3, :for_question_answering},
217+
"SmolLM3ForSequenceClassification" => {Bumblebee.Text.SmolLm3, :for_sequence_classification},
218+
"SmolLM3ForTokenClassification" => {Bumblebee.Text.SmolLm3, :for_token_classification},
219219
"SwinModel" => {Bumblebee.Vision.Swin, :base},
220220
"SwinForImageClassification" => {Bumblebee.Vision.Swin, :for_image_classification},
221221
"T5Model" => {Bumblebee.Text.T5, :base},

lib/bumblebee/text/smollm3.ex

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
defmodule Bumblebee.Text.SmolLM3 do
1+
defmodule Bumblebee.Text.SmolLm3 do
22
alias Bumblebee.Shared
33

44
options =

mix.exs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,7 @@ defmodule Bumblebee.MixProject do
108108
Bumblebee.Text.Phi,
109109
Bumblebee.Text.Phi3,
110110
Bumblebee.Text.Roberta,
111-
Bumblebee.Text.SmolLM3,
111+
Bumblebee.Text.SmolLm3,
112112
Bumblebee.Text.T5,
113113
Bumblebee.Vision.BlipVision,
114114
Bumblebee.Vision.ClipVision,

test/bumblebee/text/smollm3_test.exs

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
defmodule Bumblebee.Text.SmolLM3Test do
1+
defmodule Bumblebee.Text.SmolLm3Test do
22
use ExUnit.Case, async: true
33

44
import Bumblebee.TestHelpers
@@ -11,7 +11,7 @@ defmodule Bumblebee.Text.SmolLM3Test do
1111
architecture: :base
1212
)
1313

14-
assert %Bumblebee.Text.SmolLM3{architecture: :base} = spec
14+
assert %Bumblebee.Text.SmolLm3{architecture: :base} = spec
1515

1616
inputs = %{
1717
"input_ids" => Nx.tensor([[10, 20, 30, 40, 50, 60, 70, 80, 0, 0]]),
@@ -41,7 +41,7 @@ defmodule Bumblebee.Text.SmolLM3Test do
4141
architecture: :for_question_answering
4242
)
4343

44-
assert %Bumblebee.Text.SmolLM3{architecture: :for_question_answering} = spec
44+
assert %Bumblebee.Text.SmolLm3{architecture: :for_question_answering} = spec
4545

4646
inputs = %{
4747
"input_ids" => Nx.tensor([[10, 20, 30, 40, 50, 60, 70, 80, 0, 0]]),
@@ -78,7 +78,7 @@ defmodule Bumblebee.Text.SmolLM3Test do
7878
architecture: :for_sequence_classification
7979
)
8080

81-
assert %Bumblebee.Text.SmolLM3{architecture: :for_sequence_classification} = spec
81+
assert %Bumblebee.Text.SmolLm3{architecture: :for_sequence_classification} = spec
8282

8383
inputs = %{
8484
"input_ids" => Nx.tensor([[10, 20, 30, 40, 50, 60, 70, 80, 0, 0]]),
@@ -101,7 +101,7 @@ defmodule Bumblebee.Text.SmolLM3Test do
101101
architecture: :for_causal_language_modeling
102102
)
103103

104-
assert %Bumblebee.Text.SmolLM3{architecture: :for_causal_language_modeling} = spec
104+
assert %Bumblebee.Text.SmolLm3{architecture: :for_causal_language_modeling} = spec
105105

106106
inputs = %{
107107
"input_ids" => Nx.tensor([[10, 20, 30, 40, 50, 60, 70, 80, 0, 0]]),
@@ -131,7 +131,7 @@ defmodule Bumblebee.Text.SmolLM3Test do
131131
architecture: :for_token_classification
132132
)
133133

134-
assert %Bumblebee.Text.SmolLM3{architecture: :for_token_classification} = spec
134+
assert %Bumblebee.Text.SmolLm3{architecture: :for_token_classification} = spec
135135

136136
inputs = %{
137137
"input_ids" => Nx.tensor([[10, 20, 30, 40, 50, 60, 70, 80, 0, 0]]),

0 commit comments

Comments
 (0)