Downloads · 30 days
10
13% of all-time downloads
FlorianVal/DSSD-Llama3-8B
DSSD-Llama3-8B is a machine learning model from FlorianVal. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers.
This repository contains the trained auxiliary early-exit heads for meta-llama/Meta-Llama-3-8B. These heads are used for Dynamic Self-Speculative Decoding (DSSD), enabling faster inference with guaranteed identical ou…
Downloads · 30 days
10
13% of all-time downloads
All-time downloads
80
Public
Repo size
3.2 GB
Likes
0
Public
Click a slice to open those files.
.pt3.2 GB · 100%
From the Hugging Face model README
This repository contains the trained auxiliary early-exit heads for meta-llama/Meta-Llama-3-8B. These heads are used for Dynamic Self-Speculative Decoding (DSSD), enabling faster inference with guaranteed identical output to the full model.
These heads are designed to be used with the DSSD Demo.
aux_heads.pt: State dictionary of the trained auxiliary heads (Layers 8, 16, 24).config.json: Configuration file specifying head layers and parameters.calibration.json: Calibrated thresholds for different accuracy levels.The auxiliary heads allow the model to exit early at intermediate layers if confidence thresholds are met. If the drafted tokens are not verified by the full model, they are corrected, ensuring 100% output match.