Default to write_file prompt for stage1
This commit is contained in:
@@ -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"))
|
||||
|
||||
|
||||
Reference in New Issue
Block a user