fix peft init

1347027216 · Sep 6, 2023 · 73f44c4 · 73f44c4
1 parent f7f1400
commit 73f44c4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/scripts/training/peft/__init__.py b/scripts/training/peft/__init__.py
@@ -47,7 +47,7 @@
     TaskType,
     bloom_model_postprocess_past_key_value,
     get_peft_model_state_dict,
-    prepare_model_for_int8_training,
+    # prepare_model_for_int8_training,
     set_peft_model_state_dict,
     shift_tokens_right,
 )
diff --git a/scripts/training/peft/utils/__init__.py b/scripts/training/peft/utils/__init__.py
@@ -23,7 +23,7 @@
     TRANSFORMERS_MODELS_TO_PREFIX_TUNING_POSTPROCESS_MAPPING,
     _set_trainable,
     bloom_model_postprocess_past_key_value,
-    prepare_model_for_int8_training,
+    # prepare_model_for_int8_training,
     shift_tokens_right,
     transpose,
 )