Text Generation
Transformers
Safetensors
mistral
conversational
text-generation-inference

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
5
Safetensors
Model size
2.88B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for wandb/pruned_mistral

Finetuned
(872)
this model

Datasets used to train wandb/pruned_mistral