diff --git a/scripts/stage1_debug.py b/scripts/stage1_debug.py index 9055a32..9a624c3 100644 --- a/scripts/stage1_debug.py +++ b/scripts/stage1_debug.py @@ -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"))