language: ru
tags:
- exbert
license: mit
RuGPT2_Gen_Comments
Предварительно обученная модель на русском языке с использованием языковой модели "sberbank-ai/rugpt3small_based_on_gpt2".
Содержимое карты этой модели было создано, чтобы дополнить предоставленную информацию и привести конкретные примеры её использования.
Описание модели
RuGPT2_Gen_Comments — это модель преобразователей, предварительно обученная на массиве данных Lenta2 проекта CORUS на русском языке в режиме самоконтроля. Этот означает, что он был предварительно обучен только необработанным текстам, и люди не маркировали их каким-либо образом (поэтому он может использовать много общедоступных данных) с автоматическим процессом создания входных данных и меток из этих текстов. Точнее, его обучали угадывать следующее слово в предложении.
Входные данные — это последовательности непрерывного текста определенной длины (block_size = 1048).
Проимер использования
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("Dmitriy007/rugpt2_gen_comments") model = AutoModelForCausalLM.from_pretrained("Dmitriy007/rugpt2_gen_comments")
input_text = 'Ученик старшего класса лицея № 21 Иван Сидоров из города Адлер полетел в космос на планету Марс.'
inputs = tokenizer(input_text, return_tensors="pt") model.to('cuda') inputs.to('cuda')
input_ids = inputs["input_ids"]
output = model.generate( input_ids, attention_mask=inputs["attention_mask"], pad_token_id=model.config.bos_token_id, max_length=300, num_beams=5, num_return_sequences=1, top_k=50, top_p=0.90, no_repeat_ngram_size=2, temperature=0.7, early_stopping=True )
generated_text = list(map(tokenizer.decode, output)) print(generated_text[0])