bjoernp commited on
Commit
b694477
1 Parent(s): ca9e5d4

Update convert_mistral_moe_weights_to_hf.py

Browse files
convert_mistral_moe_weights_to_hf.py CHANGED
@@ -24,8 +24,8 @@ from transformers import (
24
  LlamaTokenizer
25
  )
26
 
27
- from transformers.models.mistral.modeling_moe_mistral import MixtralForCausalLM
28
- from transformers.models.mistral.configuration_dmoe_mistral import MixtralConfig
29
 
30
  try:
31
  from transformers import LlamaTokenizerFast
 
24
  LlamaTokenizer
25
  )
26
 
27
+ from .modeling_moe_mistral import MixtralForCausalLM
28
+ from .configuration_moe_mistral import MixtralConfig
29
 
30
  try:
31
  from transformers import LlamaTokenizerFast