hpcaitech
diff --git a/‎examples/inference/benchmark_llama.py‎ renamed to ‎examples/inference/llama/benchmark_llama.py‎ b/‎examples/inference/benchmark_llama.py‎ renamed to ‎examples/inference/llama/benchmark_llama.py‎
diff --git a/‎examples/inference/benchmark_llama3.py‎ renamed to ‎examples/inference/llama/benchmark_llama3.py‎ b/‎examples/inference/benchmark_llama3.py‎ renamed to ‎examples/inference/llama/benchmark_llama3.py‎
diff --git a/‎examples/inference/llama_generation.py‎ renamed to ‎examples/inference/llama/llama_generation.py‎
Lines changed: 1 addition & 1 deletion b/‎examples/inference/llama_generation.py‎ renamed to ‎examples/inference/llama/llama_generation.py‎
Lines changed: 1 addition & 1 deletion
diff --git a/‎examples/inference/run_benchmark.sh‎ renamed to ‎examples/inference/llama/run_benchmark.sh‎ b/‎examples/inference/run_benchmark.sh‎ renamed to ‎examples/inference/llama/run_benchmark.sh‎
@@ -59,7 +59,7 @@ def infer(args):
     coordinator.print_on_master(out[0])
 
 
-# colossalai run --nproc_per_node 1 llama_gen.py -m MODEL_PATH
+# colossalai run --nproc_per_node 1 llama_generation.py -m MODEL_PATH
 if __name__ == "__main__":
     # ==============================
     # Parse Arguments