From 1ebc06b64d5e021c067103d265cfa14d7b94abaf Mon Sep 17 00:00:00 2001
From: RUEI4341 <82101769+RUEI4341@users.noreply.github.com>
Date: Fri, 7 Nov 2025 16:21:26 +0800
Subject: [PATCH] Maxtext/user/dora/fix ckpt scripts import error

* import llama_or_mistral_ckpt
---
 .../utils/ckpt_scripts/llama_mistral_mixtral_orbax_to_hf.py     | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/MaxText/utils/ckpt_scripts/llama_mistral_mixtral_orbax_to_hf.py b/src/MaxText/utils/ckpt_scripts/llama_mistral_mixtral_orbax_to_hf.py
index 8a9afe34d..a1f525d29 100644
--- a/src/MaxText/utils/ckpt_scripts/llama_mistral_mixtral_orbax_to_hf.py
+++ b/src/MaxText/utils/ckpt_scripts/llama_mistral_mixtral_orbax_to_hf.py
@@ -48,7 +48,7 @@
 from transformers import LlamaForCausalLM, MistralForCausalLM, AutoModelForCausalLM, AutoConfig
 
 from MaxText import checkpointing
-from MaxText import llama_or_mistral_ckpt
+import llama_or_mistral_ckpt
 from MaxText import max_logging
 from MaxText import maxtext_utils
 from MaxText import pyconfig