Default to write_file prompt for stage1

This commit is contained in:
2026-04-10 15:36:45 +00:00
parent 9a6fbd5797
commit f20d3bebc3

View File

@@ -11,7 +11,7 @@ import json
from vllm import LLM, SamplingParams
MODEL_PATH = os.environ.get("MODEL_PATH", "/workspace/models/SmolLM3-3B")
PROMPT_FILE = os.environ.get("PROMPT_FILE", "/workspace/prompts/smol_tool_call.txt")
PROMPT_FILE = os.environ.get("PROMPT_FILE", "/workspace/prompts/smol_write_file.txt")
MAX_TOKENS = int(os.environ.get("MAX_TOKENS", "512"))
TEMPERATURE = float(os.environ.get("TEMPERATURE", "0.0"))