Views
No views yet
Чем я хуже лошади
Курю на красной площади
Тощий, как береза в роще
Варю щи
Сейчас в дрищи
Ищу пищу
Пищи как мышка
Кури шишку
(с) блев мс 20071from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4if torch.cuda.is_available():
5 device = torch.device("cuda")
6 print("Using GPU:", torch.cuda.get_device_name(0))
7
8 model = AutoModelForCausalLM.from_pretrained("screamuch/kunteynir-no-blev-ruadapt_llama2_7b")
9 tokenizer = AutoTokenizer.from_pretrained("screamuch/kunteynir-no-blev-ruadapt_llama2_7b")
10
11 model = model.half() # вот это важно почему-то, я хз
12
13 model.to(device)1input_text = """### Input:
2
3Напиши куплет в стиле Паши Техника про шепчащийся человека, который закончил работу в больнице и хочет делиться своим вечерним опытом на пути до дома, где он направляется в кабинет уролога по талону и сталкивается с разными сценaми, включая спящую пациентку и неповиновение требованию снять куртку Stone Island и пойти на третий этаж в лабораторию. Автор использует юмор и игру слов на тему медицины и ходжефукса во время этого рассказа.
4
5### Response:
6"""
7input_ids = tokenizer.encode(input_text, return_tensors="pt")
8
9input_ids = input_ids.to(device)
10
11output_ids = model.generate(input_ids, max_length=500, num_beams=2, early_stopping=True)
12
13generated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
14print(generated_text)Напиши куплет в стиле Паши Техника/Кальмара/Максима Синицына/Блева про для более качественного ответа### Input:
Напиши куплет в стиле Паши Техника про лысого человека, который ищет шишки, а также упоминает Джунгли (возможно, название места или группы), Маугли Багира и Иру в связке с второй ночью, во время которой они веселятся с чипсами и сыром. Автор также находится в атмосфере дыма от трубы и пива.
### Response:
О чём говорим, о чём пишем
Лысый, не тупи, доставай шиши
Извини, это все, что я могу сказать
Слышь, Юру, Иру, держи его за яйца
Вторую ночь под порохом ебашим
Чипсы с сыром, это всегда круто
Под дымом куря опиум, пиво выкуриваю
Напоминаю, что ты был с нами в Джунглях
Мы тебя не забыли
В лесу были мы, Маугли и Багира
Наш дом утек в океан
Давно не видел набухший хуй
Но сейчас у меня бомба в кармане
И свет, и газ
### End: