Model card: INSIGHT Lab affiliations and Colab link

README.md

@@ -17,13 +17,14 @@ tags:
 
 **On the SPOT: Enhancing Speculative Decoding for Diffusion Language Models via Commit-Order Prediction**
 
-Eliezer Shlomi<sup>1</sup> · Eliron Rahimi<sup>1</sup> · Avi Mendelson<sup>1</sup> · Chaim Baskin<sup>2,3</sup>
+Eliezer Shlomi<sup>1,2</sup> · Eliron Rahimi<sup>1,2</sup> · Avi Mendelson<sup>1</sup> · Chaim Baskin<sup>2,3</sup>
 
-<sup>1</sup>Technion - Israel Institute of Technology &nbsp; <sup>2</sup>Decart AI &nbsp; <sup>3</sup>Ben Gurion University of the Negev
+<sup>1</sup>Technion - Israel Institute of Technology &nbsp; <sup>2</sup>INSIGHT Lab, School of Electrical and Computer Engineering, Ben-Gurion University of the Negev &nbsp; <sup>3</sup>Decart AI
 
-[![Project page](https://img.shields.io/badge/Project-page-C43D2A)](https://eliezer318.github.io/SPOT-dLLM/)
+[![Project page](https://img.shields.io/badge/Project-page-1D4ED8)](https://eliezer318.github.io/SPOT-dLLM/)
 [![Paper](https://img.shields.io/badge/Paper-coming%20soon-lightgrey)](#citation)
 [![Code](https://img.shields.io/badge/GitHub-SPOT--dLLM-181717?logo=github)](https://github.com/Eliezer318/SPOT-dLLM)
+[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/Eliezer318/SPOT-dLLM/blob/main/colab/spot_quickstart.ipynb)
 [![License](https://img.shields.io/badge/license-MIT-blue)](https://github.com/Eliezer318/SPOT-dLLM/blob/main/LICENSE)
 
 </div>
@@ -60,7 +61,7 @@ The checkpoints are loaded by the decoder in the [GitHub repository](https://git
 git clone https://github.com/Eliezer318/SPOT-dLLM.git && cd SPOT-dLLM
 pip install torch --index-url https://download.pytorch.org/whl/cu124   # match your driver
 pip install -r requirements.txt
-huggingface-cli download EliezerS/SPOT-dLLM --local-dir drafters
+huggingface-cli download HF_REPO_ID --local-dir drafters
 python run_sweep.py --model sdar --block-length 32 --gen-length 256 --task gsm8k --limit 100 \
   --temperature 0.0 --verify match --layout concat --cache on --unmasking single --suffix skip \
   --pad-pieces -1 --draft slot --slot-mode ddtree_b --slot-ckpt drafters/sdar8b_b32.pt \