We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 70f07c9 + ee9f077 commit 9655271Copy full SHA for 9655271
mergekit/moe/mixtral.py
@@ -9,7 +9,7 @@
9
import transformers
10
11
from mergekit.architecture import WeightInfo
12
-from mergekit.architecture.mixtral import MISTRAL_INFO
+from mergekit.architecture.moe_defs import MISTRAL_INFO
13
from mergekit.moe.arch import MoEOutputArchitecture
14
from mergekit.moe.common import copy_tensor_out, initialize_io, select_dtype
15
from mergekit.moe.config import MoEMergeConfig
0 commit comments