Correct import paths

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

Author: danielkorzekwa

modelopt/torch/_compress/nas/plugins/compress_nas_plugin.py

@@ -26,9 +26,11 @@
 import score_pruning_activations
 import scoring
 import torch
-from scripts.convert_llama3_to_decilm import convert_llama3_to_decilm
 from torch import nn
 
+from modelopt.torch._compress.decilm.converters.convert_llama3_to_decilm import (
+    convert_llama3_to_decilm,
+)
 from modelopt.torch._compress.runtime import NativeDdpRuntime
 from modelopt.torch.nas.conversion import NASModeRegistry
 from modelopt.torch.opt.config import ModeloptBaseConfig, ModeloptField

tests/experimental/torch/_compress/test_compress.py

@@ -27,9 +27,11 @@
     setup_puzzle_dir,
 )
 from puzzle_tools.hydra_utils import register_hydra_resolvers
-from scripts.convert_llama3_to_decilm import convert_llama3_to_decilm
 
 from modelopt.torch._compress import compress
+from modelopt.torch._compress.decilm.converters.convert_llama3_to_decilm import (
+    convert_llama3_to_decilm,
+)
 from modelopt.torch._compress.runtime import NativeDdpRuntime
 
 # The e2e test to compress a model based on Local Neural Architecture Search (Mixed Integer Programing NAS search)