version bumped

Author: DzmitryPihulski

examples/example_finetune_args.yaml

@@ -2,7 +2,7 @@
 LLMSQL — A Text2SQL benchmark for evaluation of Large Language Models
 """
 
-__version__ = "0.1.6"
+__version__ = "0.1.7"
 
 
 def __getattr__(name: str):  # type: ignore

examples/finetune.sh

@@ -1,31 +1,12 @@
 import argparse
 import inspect
 import json
-import sys
 
 
 def main() -> None:
     parser = argparse.ArgumentParser(prog="llmsql", description="LLMSQL CLI")
     subparsers = parser.add_subparsers(dest="command")
 
-    # ================================================================
-    # Fine-tuning command
-    # ================================================================
-    ft_parser = subparsers.add_parser(
-        "finetune",
-        help="Fine-tune a causal LM on the LLMSQL benchmark.",
-        description="Launch fine-tuning using Hugging Face TRL's SFTTrainer.\n\n"
-        "You can pass parameters directly on the CLI or provide a YAML "
-        "config file via --config-file. CLI args take precedence.",
-        epilog="Example:\n  llmsql finetune --config-file examples/example_finetune_args.yaml",
-    )
-    ft_parser.add_argument(
-        "--config-file",
-        type=str,
-        required=True,
-        help="Path to a YAML config file containing training parameters.",
-    )
-
     # ================================================================
     # Inference command
     # ================================================================
@@ -88,19 +69,10 @@ def main() -> None:
     # ================================================================
     args, extra = parser.parse_known_args()
 
-    # ------------------------------------------------
-    # Fine-tune
-    # ------------------------------------------------
-    if args.command == "finetune":
-        from llmsql.finetune import finetune
-
-        sys.argv = ["llmsql-finetune"] + extra + ["--config_file", args.config_file]
-        finetune.run_cli()
-
     # ------------------------------------------------
     # Inference
     # ------------------------------------------------
-    elif args.command == "inference":
+    if args.command == "inference":
         if args.method == "vllm":
             from llmsql import inference_vllm as inference_fn
         elif args.method == "transformers":

llmsql/__init__.py

@@ -4,7 +4,7 @@ build-backend = "pdm.backend"
 
 [project]
 name = "llmsql"
-version = "0.1.6"
+version = "0.1.7"
 description = "A Text2SQL benchmark for evaluation of Large Language Models"
 keywords = ["llm", "sql", "benchmark", "evaluation", "ai", "nlp"]
 authors = [