import refactoring

Signed-off-by: Daniel Korzekwa <[email protected]>

Author: danielkorzekwa

modelopt/torch/_compress/subblock_stats/calc_subblock_memory.py

@@ -20,9 +20,9 @@
 import numpy as np
 import torch
 from puzzle_tools.deci_lm_hf_code.block_config import AttentionConfig, FFNConfig, MambaConfig
-from puzzle_tools.deci_lm_hf_code.configuration_decilm import DeciLMConfig
-from puzzle_tools.deci_lm_hf_code.modeling_decilm import DeciLMMoe
 
+from modelopt.torch._compress.decilm.deci_lm_hf_code.configuration_decilm import DeciLMConfig
+from modelopt.torch._compress.decilm.deci_lm_hf_code.modeling_decilm import DeciLMMoe
 from modelopt.torch._compress.utils.utils import (
     calculate_kv_dim,
     raise_unknown_subblock_config_error,