Locutusque's picture
Update README.md
de9849a verified
|
raw
history blame
1.63 kB
metadata
license: apache-2.0
language:
  - en
  - code
datasets:
  - open-phi/programming_books_llama
  - open-phi/textbooks
tags:
  - merge
  - computer science
inference:
  parameters:
    do_sample: true
    temperature: 0.2
    top_p: 0.14
    top_k: 12
    max_new_tokens: 250
    repetition_penalty: 1.15
widget:
  - text: 'To calculate the factorial of n, we can use the following function:'

TinyMistral-248M-v2.5

This model was created by merging TinyMistral-248M-v1 and v2, then further pretraining on synthetic textbooks. The resulting model's performance is superior to both, after personal evaluation.

During training, this model reached an average perplexity score of 4, outperforming V1 by nearly 7x, and V2 by 4x.

You can use the following config to reproduce the merged model:

base_model: Locutusque/TinyMistral-248M-v2
dtype: float16
merge_method: ties
parameters:
  int8_mask: 1.0
  normalize: 1.0
slices:
- sources:
  - layer_range: [0, 12]
    model: Locutusque/TinyMistral-248M
    parameters:
      density: [1.0, 0.7, 0.1]
      weight: 1.0
  - layer_range: [0, 12]
    model: Locutusque/TinyMistral-248M-v2
    parameters:
      density: 0.5
      weight: [0.0, 0.3, 0.7, 1.0]

This model can also answer basic questions, without needing to do any fine-tuning.

This model was also created as an attempt to fix the issue with V2 - the weights were prone to exploding gradients, making it difficult to fine-tune. This model is easier to fine-tune.

To get the best out of this model, I recommend installing it, and trying it out yourself, as the model's performance seems to degrade in the inference API.