additional fixes

training/scripts/inference.py

@@ -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")

training/scripts/run_with_accelerate.sh

@@ -7,4 +7,5 @@ set -e
 pip install -q accelerate
 
 # Launch with accelerate
-accelerate launch train-on-this-server.sh --config training/configs/llama2-7b-lora.yaml
+# Note: Use the dedicated train-on-this-server.sh instead
+echo "See train-on-this-server.sh for full setup"