update moudle name

Author: jitingxu1

distill_step_by_step.py

@@ -25,7 +25,7 @@
 from transformers import AutoTokenizer, T5ForConditionalGeneration
 
 from model_utils import TaskPrefixTrainer, TaskPrefixDataCollator
-from model import FIM_PREFIX, FIM_MIDDLE, FIM_SUFFIX
+from santacoder_model import FIM_PREFIX, FIM_MIDDLE, FIM_SUFFIX
 from metrics import compute_metrics_text