Skip to content

Commit 435f88f

Browse files
authored
nit: typing use Llama4TextConfig instead of Llama4Config (#37430)
nit: typing to text config Signed-off-by: Mehant Kammakomati <mehant.kammakomati2@ibm.com>
1 parent 954f31c commit 435f88f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/transformers/models/llama4/modeling_llama4.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@
6060

6161

6262
class Llama4TextExperts(nn.Module):
63-
def __init__(self, config: Llama4Config):
63+
def __init__(self, config: Llama4TextConfig):
6464
super().__init__()
6565
self.num_experts = config.num_local_experts
6666
self.intermediate_size = config.intermediate_size

0 commit comments

Comments
 (0)