cmedina/agenx-lora-training
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

additional fixes

Browse Source
This commit is contained in:
Christian Medina
2026-07-01 16:05:52 -04:00
parent 088521094e
commit 7e13680e27
4 changed files with 13 additions and 88 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
training/scripts/inference.py Executable file → Normal file
View File

@@ -68,8 +68,8 @@ def generate_summary(model, tokenizer, task, files_changed=None, tests_run=False
def main():
parser = argparse.ArgumentParser(description="Generate Cyron summaries with LoRA")
parser.add_argument("--model", type=str, default="output/llama2-7b-lora",
help="Path to trained model")
parser.add_argument("--model", type=str, default="output/ornith-35b-lora-cyron",
help="Path to trained LoRA model")
parser.add_argument("--task", type=str, required=True, help="Task description")
parser.add_argument("--files", type=str, nargs="*", default=[], help="Files changed")
parser.add_argument("--tests-run", action="store_true", help="Tests were run")