Skip to content

Commit 3e060e1

Browse files
author
Yunchao Yang
committed
[gemma4] Fix isort/black lint errors in __init__.py and tokenizer.py
1 parent 159c833 commit 3e060e1

2 files changed

Lines changed: 6 additions & 2 deletions

File tree

src/fairseq2/models/gemma4/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,13 +24,13 @@
2424
)
2525
from fairseq2.models.gemma4.config import GEMMA4_FAMILY as GEMMA4_FAMILY
2626
from fairseq2.models.gemma4.config import Gemma4Config as Gemma4Config
27+
from fairseq2.models.gemma4.config import get_gemma4_12b_config as get_gemma4_12b_config
2728
from fairseq2.models.gemma4.config import (
2829
get_gemma4_26b_a4b_config as get_gemma4_26b_a4b_config,
2930
)
3031
from fairseq2.models.gemma4.config import get_gemma4_31b_config as get_gemma4_31b_config
3132
from fairseq2.models.gemma4.config import get_gemma4_e2b_config as get_gemma4_e2b_config
3233
from fairseq2.models.gemma4.config import get_gemma4_e4b_config as get_gemma4_e4b_config
33-
from fairseq2.models.gemma4.config import get_gemma4_12b_config as get_gemma4_12b_config
3434
from fairseq2.models.gemma4.config import (
3535
register_gemma4_configs as register_gemma4_configs,
3636
)
@@ -84,6 +84,7 @@
8484
"apply_fsdp_to_gemma4",
8585
"convert_gemma4_state_dict",
8686
"create_gemma4_model",
87+
"get_gemma4_12b_config",
8788
"get_gemma4_26b_a4b_config",
8889
"get_gemma4_31b_config",
8990
"get_gemma4_e2b_config",

src/fairseq2/models/gemma4/tokenizer.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,10 @@ def create_encoder(
5858
raise ValueError(f"`lang` must be `None`, but is '{lang}' instead.")
5959

6060
if mode is not None and mode not in (
61-
"default", "prompt", "prompt_response", "as_is"
61+
"default",
62+
"prompt",
63+
"prompt_response",
64+
"as_is",
6265
):
6366
raise ValueError(
6467
"`mode` must be 'default', 'prompt', 'prompt_response', or "

0 commit comments

Comments
 (0)