Default to write_file prompt for stage1
This commit is contained in:
@@ -11,7 +11,7 @@ import json
|
|||||||
from vllm import LLM, SamplingParams
|
from vllm import LLM, SamplingParams
|
||||||
|
|
||||||
MODEL_PATH = os.environ.get("MODEL_PATH", "/workspace/models/SmolLM3-3B")
|
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"))
|
MAX_TOKENS = int(os.environ.get("MAX_TOKENS", "512"))
|
||||||
TEMPERATURE = float(os.environ.get("TEMPERATURE", "0.0"))
|
TEMPERATURE = float(os.environ.get("TEMPERATURE", "0.0"))
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user