Skip to content

Commit a37bba3

Browse files
committed
move Llama4Config importation path for compatibility with wider transformers versions
Signed-off-by: cjackal <44624812+cjackal@users.noreply.github.com>
1 parent 10a3cc2 commit a37bba3

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/llmcompressor/modeling/llama4.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,10 @@
11
from typing import Tuple
22

33
import torch
4-
from transformers.models import Llama4Config
5-
from transformers.models.llama4.configuration_llama4 import Llama4TextConfig
4+
from transformers.models.llama4.configuration_llama4 import (
5+
Llama4Config,
6+
Llama4TextConfig,
7+
)
68
from transformers.models.llama4.modeling_llama4 import (
79
Llama4TextExperts,
810
Llama4TextMLP,

0 commit comments

Comments
 (0)