We introduce Wavefront Decoding (WFD), a training-free, lossless self-speculative decoding framework native to looped language models.
Make sure you pulled submodules
git submodule update --init
We recommend creating a virtual environment and installing PyTorch first.
conda create -n [name] python=3.11 -y
conda activate [name]
pip install torch
pip install -r env/requirements.txt
python env/verify_env.py
CUDA_VISIBLE_DEVICES=[gpu_num] bash experiments/exp2_spec_bench/run.sh
Edit run.sh to configure options before running.
CUDA_VISIBLE_DEVICES=[gpu_num] bash experiments/exp3_accuracy/run.sh
Edit run.sh to configure options before running.
CUDA_VISIBLE_DEVICES=[gpu_num] bash experiments/exp4_emu/run.sh
Edit run.sh to configure options before running.
CUDA_VISIBLE_DEVICES=[gpu_num] bash experiments/exp5_graph/run.sh
Edit run.sh to configure options before running.
We introduce Wavefront Decoding (WFD), a training-free, lossless self-speculative decoding framework native to looped language models.
Make sure you pulled submodules
git submodule update --init
We recommend creating a virtual environment and installing PyTorch first.
conda create -n [name] python=3.11 -y
conda activate [name]
pip install torch
pip install -r env/requirements.txt
python env/verify_env.py
CUDA_VISIBLE_DEVICES=[gpu_num] bash experiments/exp2_spec_bench/run.sh
Edit run.sh to configure options before running.
CUDA_VISIBLE_DEVICES=[gpu_num] bash experiments/exp3_accuracy/run.sh
Edit run.sh to configure options before running.
CUDA_VISIBLE_DEVICES=[gpu_num] bash experiments/exp4_emu/run.sh
Edit run.sh to configure options before running.
CUDA_VISIBLE_DEVICES=[gpu_num] bash experiments/exp5_graph/run.sh
Edit run.sh to configure options before running.