diff --git a/offsite_tuning/run_clm.py b/offsite_tuning/run_clm.py
index 37b9d67..b3686d7 100644
--- a/offsite_tuning/run_clm.py
+++ b/offsite_tuning/run_clm.py
@@ -68,7 +68,6 @@
     add_epilogue,
     add_prologue,
     uniform_choose_layers,
-    choose_layers_by_changes,
     magnitude_prune,
     quantize,
     parse_args,