README.md

metadata

base_model: AI-Sweden-Models/gpt-sw3-6.7b-v2-instruct
language:
  - sv
  - da
  - 'no'
  - en
pipeline_tag: text-generation
tags:
  - translation

Model Card for gpt-sw3-6.7b-v2-translator-gguf

The gpt-sw3-6.7b-v2-translator is a finetuned version of gpt-sw3-6.7b-v2-instruct on a carefully selected translation pair dataset that was gathered by AI Sweden.

Intended usage:

Translate text data from English to Swedish, or Swedish to English.

How to use:

Translate from English to Swedish:

FROM ./gpt-sw3-6-7b-v2-translator-SIZE.gguf
TEMPLATE "<|endoftext|><s>User: Översätt till Svenska från Engelska\n{{ .Prompt }}<s>Bot:"
PARAMETER stop <s>

Translate from Swedish to English:

FROM ./gpt-sw3-6-7b-v2-translator-SIZE.gguf
TEMPLATE "<|endoftext|><s>User: Översätt till Engelska från Svenska\n{{ .Prompt }}<s>Bot:"
PARAMETER stop <s>

Versions:

gpt-sw3-6-7b-v2-translator-q4.gguf
gpt-sw3-6-7b-v2-translator-q8.gguf
gpt-sw3-6-7b-v2-translator-f16.gguf
gpt-sw3-6-7b-v2-translator-f32.gguf

Training & Data:

The training was done on 1 NVIDIA DGX using DeepSpeed ZeRO 3 for three epochs on roughly 4GB of carefully selected translation data. It is a full finetune of all of the model parameters.

Epoch	Training Loss	Evaluation Loss
1	1.309	1.281
2	1.161	1.242
3	1.053	1.219