iolai-2026-4b-wvote / script.py
richardlian's picture
Upload folder using huggingface_hub
010384c verified
Raw
History Blame Contribute Delete
1.36 kB
"""IOL-AI 2026 submission: q4b-wvote-v1.
E33 confidence-weighted votes: 12 samples, per-item raw-model logprob weighting (T=0.35).
Recipe (every knob is read by v2/config.py::Config.from_env):
BACKEND=parallel
FEWSHOT=0
LLAMA_CTX=16384
MAX_NEW=700
N_SAMPLES=12
PROMPT_MODE=direct
REFINE_ROUNDS=0
SLOTS=16
THINK=0
TRANSCRIPT=
VOTE_LP_T=0.35
VOTE_TEMP=1.0
VOTE_TOP_P=0.95
VOTE_WEIGHT=full
"""
import os
import sys
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
HERE = os.path.dirname(os.path.abspath(__file__))
os.environ.setdefault("GGUF_PATH", os.path.join(HERE, "model", "model-Q8_0.gguf"))
os.environ.setdefault("LLAMA_PARALLEL", os.path.join(HERE, "bin", "llama-iolbatch"))
os.environ.setdefault("BACKEND", "parallel")
os.environ.setdefault("SLOTS", "16")
os.environ.setdefault("LLAMA_CTX", "16384")
os.environ.setdefault("VOTE_TEMP", "1.0")
os.environ.setdefault("VOTE_TOP_P", "0.95")
os.environ.setdefault("PROMPT_MODE", "direct")
os.environ.setdefault("N_SAMPLES", "12")
os.environ.setdefault("MAX_NEW", "700")
os.environ.setdefault("FEWSHOT", "0")
os.environ.setdefault("THINK", "0")
os.environ.setdefault("REFINE_ROUNDS", "0")
os.environ.setdefault("TRANSCRIPT", "")
os.environ.setdefault("VOTE_WEIGHT", "full")
os.environ.setdefault("VOTE_LP_T", "0.35")
from v2.script import main # noqa: E402
main()