kulyk-en-uk / README.md
Yehor's picture
Update README.md
a77db85 verified
metadata
model-index:
  - name: Kulyk-EN-UK
    results:
      - task:
          type: text-generation
        dataset:
          type: facebook/flores
          name: FLORES
          split: devtest
        metrics:
          - type: bleu
            value: 27.24
            name: BLEU
library_name: transformers
license: other
license_name: lfm1.0
license_link: LICENSE
language:
  - en
  - uk
pipeline_tag: text-generation
tags:
  - liquid
  - lfm2
  - edge
datasets:
  - lang-uk/FiftyFiveShades
base_model:
  - LiquidAI/LFM2-350M

A new lightweight model to do machine translation from English to Ukrainian using recently published LFM2 model. Use demo to test it.

Facts:

  • Fine-tuned with 40M samples (filtered by quality metric) from ~53.5M for 1.4 epochs
  • 354M params
  • Requires 1 GB of RAM to run with bf16
  • BLEU on FLORES-200: 27.24
  • Tokens per second: 229.93 (bs=1), 1664.40 (bs=10), 8392.48 (bs=64)
  • License: lfm1.0

Info:

  • Model name is inherited from name of Sergiy Kulyk who was chargé d'affaires of Ukraine in the United States

Training Info:

  • Learning Rate: 3e-5
  • Learning Rate scheduler type: cosine
  • Warmup Ratio: 0.05
  • Max length: 2048
  • packed=True
  • Sentences <= 1000 chars
  • Gradient accumulation steps: 4
  • Used Flash Attention 2
  • Time for epoch: 32 hours
  • 2 cards of NVIDIA RTX 3090 Ti (24G)
  • accelerate with DeepSpeed, offloading into CPU
  • Memory usage: 22.212GB-22.458GB
  • torch 2.7.1

Acknowledgements:

  • Serhiy Stetskovych for providing compute to train this model
  • lang-uk members for their compilation of different MT datasets