spivavtor-large / README.md
amansaini's picture
Update README.md (#3)
4607a6a verified
|
raw
history blame
2.76 kB
metadata
license: cc-by-nc-4.0
language:
  - uk
datasets:
  - grammarly/spivavtor
widget:
  - text: >-
      Перефразуйте речення: Який найкращий комплiмент, який ти отримував вiд
      будь-кого?
    example_title: Paraphrasing example
  - text: 'Спростiть речення: Там він помер через шість тижнів, 13 січня 888 року.'
    example_title: Simplification example
  - text: >-
      Виправте граматику в цьому реченнi: Дякую за інформацію! ми з Надією саме
      вийшли з дому
    example_title: GEC example
  - text: >-
      Виправте зв’язнiсть в реченнi: Лінч досі відмовляється розповідати про
      сумнозвісну травневу дорожньо-транспортну пригоду, коли він збив
      жінку-пішохода в районі нічного клубу Баффало та відлетів. Той факт, що
      Лінч взагалі говорив, заслуговує на увагу в цих краях.
    example_title: Coherence example
inference:
  parameters:
    max_new_tokens: 250

Model Card for Spivavtor-Large

This model was obtained by instruction tuning bigscience/mt0-large model on the Spivavtor dataset. All details of the dataset and fine tuning process can be found in our paper.

Paper: Spivavtor: An Instruction Tuned Ukrainian Text Editing Model

Authors: Aman Saini, Artem Chernodub, Vipul Raheja, Vivek Kulkarni

Model Details

Model Description

  • Language: Ukrainian
  • Finetuned from model: bigscience/mt0-large

How to use

We make the following models available from our paper.

Model Number of parameters Reference name in Paper
Spivavtor-large 1.2B SPIVAVTOR-MT0-LARGE
Spivavtor-xxl 11B SPIVAVTOR-AYA-101

Usage

from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("grammarly/spivavtor-large")
model = AutoModelForSeq2SeqLM.from_pretrained("grammarly/spivavtor-large")

# Paraphrase the sentence: What is the greatest compliment that you ever received from anyone?
input_text = 'Перефразуйте речення: Який найкращий комплімент, який ти отримував від будь-кого?'

inputs = tokenizer.encode(input_text, return_tensors="pt")
output = model.generate(inputs, max_length=256)
output_text = tokenizer.decode(outputs[0], skip_special_tokens=True)