Skip to content

Commit 8c67cf7

Browse files
Add mip module
Signed-off-by: Daniel Korzekwa <[email protected]>
1 parent 4899bb5 commit 8c67cf7

File tree

8 files changed

+1150
-19
lines changed

8 files changed

+1150
-19
lines changed

modelopt/torch/_compress/compress.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -20,15 +20,15 @@
2020
2121
"""
2222

23-
import build_library_and_stats
24-
import mip_and_realize_models
25-
import pruning_ckpts
26-
import score_pruning_activations
27-
import scoring
2823
from omegaconf import DictConfig
29-
from puzzle_tools.runtime import IRuntime
3024

25+
import modelopt.torch._compress.activation_scoring.score_pruning_activations as score_pruning_activations
26+
import modelopt.torch._compress.build_library_and_stats as build_library_and_stats
27+
import modelopt.torch._compress.mip.mip_and_realize_models as mip_and_realize_models
28+
import modelopt.torch._compress.pruning.pruning_ckpts as pruning_ckpts
29+
import modelopt.torch._compress.scoring.scoring as scoring
3130
from modelopt.torch._compress.tools.hydra_utils import initialize_hydra_config_for_dir
31+
from modelopt.torch._compress.tools.runtime import IRuntime
3232

3333

3434
def compress(

0 commit comments

Comments
 (0)