add inf bash

Xianchao-Wu · Aug 10, 2023 · 8c4cc7e · 8c4cc7e
1 parent d7527ab
commit 8c4cc7e
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/WizardCoder/inf.sh b/WizardCoder/inf.sh
@@ -0,0 +1,16 @@
+#########################################################################
+# File Name: inf.sh
+# Author: Xianchao Wu
+# mail: [email protected]
+# Created Time: Mon Jul  3 04:13:18 2023
+#########################################################################
+#!/bin/bash
+
+ckpt="/workspace/asr/Llama-X/src/checkpoints_wcode/models--WizardLM--WizardCoder-15B-V1.0/snapshots/69e87732535159460155972c3fac32a6241cc0ca"
+indata="/workspace/asr/WizardLM/WizardCoder/data/in.data.jsonl"
+outdata="/workspace/asr/WizardLM/WizardCoder/data/out.res.jsonl"
+
+python -m ipdb src/inference_wizardcoder.py \
+    --base_model $ckpt \
+    --input_data_path $indata \
+    --output_data_path $outdata