add ziya prompt template

src/utils/__init__.py

@@ -1,6 +1,7 @@
 from .common import (
     load_pretrained,
     prepare_args,
+    prepare_infer_args,
     prepare_data,
     preprocess_data
 )
@@ -13,5 +14,4 @@ from .seq2seq import ComputeMetrics, Seq2SeqPeftTrainer
 from .pairwise import PairwiseDataCollatorWithPadding, PairwisePeftTrainer
 from .ppo import PPOPeftTrainer
 
-from .config import ModelArguments, FinetuningArguments
 from .other import get_logits_processor, plot_loss