From 1bed9fa4352d71cbd0649512378dcd82d6fd80c0 Mon Sep 17 00:00:00 2001 From: Abhinav Singh Date: Fri, 7 Nov 2025 18:32:03 +0000 Subject: [PATCH] Fix vllm weight mapping import issue. --- src/MaxText/integration/tunix/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/MaxText/integration/tunix/utils.py b/src/MaxText/integration/tunix/utils.py index 43ca8577c..62eb10257 100644 --- a/src/MaxText/integration/tunix/utils.py +++ b/src/MaxText/integration/tunix/utils.py @@ -16,11 +16,11 @@ import re -import maxtext.src.maxtext.integration.tunix.weight_mapping as weight_mapping # pylint: disable=consider-using-from-import +from MaxText.integration.tunix.weight_mapping import StandaloneVllmWeightMapping from MaxText.utils.ckpt_conversion.utils.param_mapping import PARAM_MAPPING from MaxText.utils.ckpt_conversion.utils.param_mapping import VLLM_HOOK_FNS -STANDALONE_VLLM_WEIGHT_MAPPING = weight_mapping.StandaloneVllmWeightMapping() +STANDALONE_VLLM_WEIGHT_MAPPING = StandaloneVllmWeightMapping() # This static map provides the architectural knowledge (sharding) that is # not present in the original HF mapping.