tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Fix import

Browse Source
This commit is contained in:
Casper
2023-12-10 17:21:06 +01:00
parent 2ac1a72e4b
commit 083beb6425
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/axolotl/models/mixtral/__init__.py
View File

@@ -3,4 +3,7 @@ Custom modeling code for mixtral
"""
from .configuration_moe_mistral import MixtralConfig # noqa
from .modeling_moe_mistral import MixtralForCausalLM # noqa
from .modeling_moe_mistral import ( # noqa
MixtralForCausalLM,
replace_mixtral_mlp_with_swiglu,
)