"
}
elif self.lang == 'en':
return {
'title': 'Meditation for creativity development',
'description': "Write an expert article on the topic: 'Meditation for creativity development'. Compare different approaches to problem solving. Write in en language"
}
else:
return {
'title': 'Медитация для развития креативности',
'description': "Напиши экспертную статью на тему: 'Медитация для развития креативности'. Сравни разные подходы к решению проблемы. Пиши на ru языке"
}
def update_question(self, question):
if self.lang == 'en':
return question.replace('напиши', 'write').replace('Напиши', 'Write')
else:
return question
class ExpertAnswers(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напишите').replace('Напиши', 'Напишите').replace('пиши', 'пишите').replace('Пиши', 'Пишите').replace('пишите', 'пишите').replace('Пишите', 'Пишите').replace('write', 'write').replace('Write', 'Write')
class Stories(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши историю').replace('Напиши', 'Напиши историю').replace('пиши', 'пиши историю').replace('Пиши', 'Пиши историю').replace('write', 'write a story').replace('Write', 'Write a story')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class StoriesEnglish(Stories):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class ChatBot(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши ответ').replace('Напиши', 'Напиши ответ').replace('пиши', 'пиши ответ').replace('Пиши', 'Пиши ответ').replace('write', 'write a response').replace('Write', 'Write a response')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class ChatBotEnglish(ChatBot):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class QnA(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши ответ').replace('Напиши', 'Напиши ответ').replace('пиши', 'пиши ответ').replace('Пиши', 'Пиши ответ').replace('write', 'write a response').replace('Write', 'Write a response')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class QnAEnglish(QnA):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class Synonyms(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши синонимы').replace('Напиши', 'Напиши синонимы').replace('пиши', 'пиши синонимы').replace('Пиши', 'Пиши синонимы').replace('write', 'write synonyms').replace('Write', 'Write synonyms')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class SynonymsEnglish(Synonyms):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class Antonyms(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши антонимы').replace('Напиши', 'Напиши антонимы').replace('пиши', 'пиши антонимы').replace('Пиши', 'Пиши антонимы').replace('write', 'write antonyms').replace('Write', 'Write antonyms')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class AntonymsEnglish(Antonyms):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class Definitions(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши определение').replace('Напиши', 'Напиши определение').replace('пиши', 'пиши определение').replace('Пиши', 'Пиши определение').replace('write', 'write a definition').replace('Write', 'Write a definition')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class DefinitionsEnglish(Definitions):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class Summaries(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши краткое изложение').replace('Напиши', 'Напиши краткое изложение').replace('пиши', 'пиши краткое изложение').replace('Пиши', 'Пиши краткое изложение').replace('write', 'write a summary').replace('Write', 'Write a summary')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class SummariesEnglish(Summaries):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class Sentiment(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши тональность').replace('Напиши', 'Напиши тональность').replace('пиши', 'пиши тональность').replace('Пиши', 'Пиши тональность').replace('write', 'write the sentiment').replace('Write', 'Write the sentiment')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class SentimentEnglish(Sentiment):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class TextClassification(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши класс').replace('Напиши', 'Напиши класс').replace('пиши', 'пиши класс').replace('Пиши', 'Пиши класс').replace('write', 'write the class').replace('Write', 'Write the class')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class TextClassificationEnglish(TextClassification):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class Grammar(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши исправленное предложение').replace('Напиши', 'Напиши исправленное предложение').replace('пиши', 'пиши исправленное предложение').replace('Пиши', 'Пиши исправленное предложение').replace('write', 'write the corrected sentence').replace('Write', 'Write the corrected sentence')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class GrammarEnglish(Grammar):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class Titles(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши заголовок').replace('Напиши', 'Напиши заголовок').replace('пиши', 'пиши заголовок').replace('Пиши', 'Пиши заголовок').replace('write', 'write the headline').replace('Write', 'Write the headline')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class TitlesEnglish(Titles):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class Code(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши код').replace('Напиши', 'Напиши код').replace('пиши', 'пиши код').replace('Пиши', 'Пиши код').replace('write', 'write the code').replace('Write', 'Write the code')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class CodeEnglish(Code):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class SQL(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши sql запрос').replace('Напиши', 'Напиши sql запрос').replace('пиши', 'пиши sql запрос').replace('Пиши', 'Пиши sql запрос').replace('write', 'write the sql query').replace('Write', 'Write the sql query')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class SQLEnglish(SQL):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class Markdown(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши в формате markdown').replace('Напиши', 'Напиши в формате markdown').replace('пиши', 'пиши в формате markdown').replace('Пиши', 'Пиши в формате markdown').replace('write', 'write in markdown format').replace('Write', 'Write in markdown format')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class MarkdownEnglish(Markdown):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class MarkdownLatex(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши в формате markdown с поддержкой latex').replace('Напиши', 'Напиши в формате markdown с поддержкой latex').replace('пиши', 'пиши в формате markdown с поддержкой latex').replace('Пиши', 'Пиши в формате markdown с поддержкой latex').replace('write', 'write in markdown format with latex support').replace('Write', 'Write in markdown format with latex support')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class MarkdownLatexEnglish(MarkdownLatex):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class MarkdownLatexEnglishWithStyling(MarkdownLatex):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши в формате markdown с поддержкой latex и стилями').replace('Напиши', 'Напиши в формате markdown с поддержкой latex и стилями').replace('пиши', 'пиши в формате markdown с поддержкой latex и стилями').replace('Пиши', 'Пиши в формате markdown с поддержкой latex и стилями').replace('write', 'write in markdown format with latex support and styles').replace('Write', 'Write in markdown format with latex support and styles')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class CodeSQL(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши код').replace('Напиши', 'Напиши код').replace('пиши', 'пиши код').replace('Пиши', 'Пиши код').replace('write', 'write the code').replace('Write', 'Write the code')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class CodeSQLEnglish(CodeSQL):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class Images(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши код').replace('Напиши', 'Напиши код').replace('пиши', 'пиши код').replace('Пиши', 'Пиши код').replace('write', 'write the code').replace('Write', 'Write the code')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class ImagesEnglish(Images):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class ImageDescriptions(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши описание').replace('Напиши', 'Напиши описание').replace('пиши', 'пиши описание').replace('Пиши', 'Пиши описание').replace('write', 'write the description').replace('Write', 'Write the description')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class ImageDescriptionsEnglish(ImageDescriptions):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class ImageQuestions(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши вопрос').replace('Напиши', 'Напиши вопрос').replace('пиши', 'пиши вопрос').replace('Пиши', 'Пиши вопрос').replace('write', 'write the question').replace('Write', 'Write the question')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class ImageQuestionsEnglish(ImageQuestions):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class ImageCaptions(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши подпись').replace('Напиши', 'Напиши подпись').replace('пиши', 'пиши подпись').replace('Пиши', 'Пиши подпись').replace('write', 'write the caption').replace('Write', 'Write the caption')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class ImageCaptionsEnglish(ImageCaptions):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class TextToCode(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши код').replace('Напиши', 'Напиши код').replace('пиши', 'пиши код').replace('Пиши', 'Пиши код').replace('write', 'write the code').replace('Write', 'Write the code')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class TextToCodeEnglish(TextToCode):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class CodeToText(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши текст').replace('Напиши', 'Напиши текст').replace('пиши', 'пиши текст').replace('Пиши', 'Пиши текст').replace('write', 'write the text').replace('Write', 'Write the text')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class CodeToTextEnglish(CodeToText):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class CodeToTextSQL(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши текст').replace('Напиши', 'Напиши текст').replace('пиши', 'пиши текст').replace('Пиши', 'Пиши текст').replace('write', 'write the text').replace('Write', 'Write the text')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class CodeToTextSQLEnglish(CodeToTextSQL):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class TextToSQL(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши sql запрос').replace('Напиши', 'Напиши sql запрос').replace('пиши', 'пиши sql запрос').replace('Пиши', 'Пиши sql запрос').replace('write', 'write the sql query').replace('Write', 'Write the sql query')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class TextToSQLEnglish(TextToSQL):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class SQLToText(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши текст').replace('Напиши', 'Напиши текст').replace('пиши', 'пиши текст').replace('Пиши', 'Пиши текст').replace('write', 'write the text').replace('Write', 'Write the text')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class SQLToTextEnglish(SQLToText):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class CodeToSQL(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши sql запрос').replace('Напиши', 'Напиши sql запрос').replace('пиши', 'пиши sql запрос').replace('Пиши', 'Пиши sql запрос').replace('write', 'write the sql query').replace('Write', 'Write the sql query')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class CodeToSQLEnglish(CodeToSQL):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class SQLToCode(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши код').replace('Напиши', 'Напиши код').replace('пиши', 'пиши код').replace('Пиши', 'Пиши код').replace('write', 'write the code').replace('Write', 'Write the code')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class SQLToCodeEnglish(SQLToCode):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class ImageToText(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши текст').replace('Напиши', 'Напиши текст').replace('пиши', 'пиши текст').replace('Пиши', 'Пиши текст').replace('write', 'write the text').replace('Write', 'Write the text')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class ImageToTextEnglish(ImageToText):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class TextToImage(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши изображение').replace('Напиши', 'Напиши изображение').replace('пиши', 'пиши изображение').replace('Пиши', 'Пиши изображение').replace('write', 'write the image').replace('Write', 'Write the image')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class TextToImageEnglish(TextToImage):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class TextToImageArt(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши искусство').replace('Напиши', 'Напиши искусство').replace('пиши', 'пиши искусство').replace('Пиши', 'Пиши искусство').replace('write', 'write the art').replace('Write', 'Write the art')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class TextToImageArtEnglish(TextToImageArt):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class TextToImageAnime(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши аниме').replace('Напиши', 'Напиши аниме').replace('пиши', 'пиши аниме').replace('Пиши', 'Пиши аниме').replace('write', 'write the anime').replace('Write', 'Write the anime')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class TextToImageAnimeEnglish(TextToImageAnime):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class TextToImageArtAnime(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши аниме').replace('Напиши', 'Напиши аниме').replace('пиши', 'пиши аниме').replace('Пиши', 'Пиши аниме').replace('write', 'write the anime').replace('Write', 'Write the anime')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class TextToImageArtAnimeEnglish(TextToImageArtAnime):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class TextToImageArtAnimeReal(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши аниме').replace('Напиши', 'Напиши аниме').replace('пиши', 'пиши аниме').replace('Пиши', 'Пиши аниме').replace('write', 'write the anime').replace('Write', 'Write the anime')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class TextToImageArtAnimeRealEnglish(TextToImageArtAnimeReal):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class TextToImageAnimeReal(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши аниме').replace('Напиши', 'Напиши аниме').replace('пиши', 'пиши аниме').replace('Пиши', 'Пиши аниме').replace('write', 'write the anime').replace('Write', 'Write the anime')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class TextToImageAnimeRealEnglish(TextToImageAnimeReal):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class TextToImageArtReal(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши искусство').replace('Напиши', 'Напиши искусство').replace('пиши', 'пиши искусство').replace('Пиши', 'Пиши искусство').replace('write', 'write the art').replace('Write', 'Write the art')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class TextToImageArtRealEnglish(TextToImageArtReal):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class TextToImageReal(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши изображение').replace('Напиши', 'Напиши изображение').replace('пиши', 'пиши изображение').replace('Пиши', 'Пиши изображение').replace('write', 'write the image').replace('Write', 'Write the image')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class TextToImageRealEnglish(TextToImageReal):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class TextToImageRealistic(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши изображение').replace('Напиши', 'Напиши изображение').replace('пиши', 'пиши изображение').replace('Пиши', 'Пиши изображение').replace('write', 'write the image').replace('Write', 'Write the image')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class TextToImageRealisticEnglish(TextToImageRealistic):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class TextToImageRealisticArt(Answers):
def __init__(self, lang='ru', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
def update_question(self, question):
return question.replace('напиши', 'напиши искусство').replace('Напиши', 'Напиши искусство').replace('пиши', 'пиши искусство').replace('Пиши', 'Пиши искусство').replace('write', 'write the art').replace('Write', 'Write the art')
def update_answer(self, answer):
return answer.replace('стоп', '').replace('stop', '')
class TextToImageRealisticArtEnglish(TextToImageRealisticArt):
def __init__(self, lang='en', max_tokens=1000, stop_seq_ids=[50256]):
super().__init__(lang, max_tokens, stop_seq_ids)
class TextToImageRealisticAnime