Edit model card

A Pruned Mistral model

This model is a pruned Mistral model re-aligned using the Zephyr Recipe

details

  • This model has 2 stages training: SFT and DPO
  • The initial model consist on selecting some layers of the mistral model to make a smaller model
  • the code can be found here: github.com/tcapelle/shear

W&B workspace

https://wandb.ai/llm_surgery/shearllama/

Downloads last month
25
Safetensors
Model size
2.88B params
Tensor type
BF16
·
Inference API
Input a message to start chatting with wandb/pruned_mistral.
This model can be loaded on Inference API (serverless).

Finetuned from

Datasets used to train wandb/pruned_mistral