Views
No views yet
1request = "Стих про программиста может быть таким:"
2
3encoded_input = tokenizer(request, return_tensors='pt', \
4 add_special_tokens=False).to('cuda:0')
5output = model.generate(
6 **encoded_input,
7 num_beams=2,
8 do_sample=True,
9 max_new_tokens=100
10)
11
12print(tokenizer.decode(output[0], skip_special_tokens=True))>>> Стих про программиста может быть таким:
Программист сидит в кресле,
Стих сочиняет он про любовь,
Он пишет, пишет, пишет, пишет...
И не выходит ни черта!1request = "Нейронная сеть — это"
2
3encoded_input = tokenizer(request, return_tensors='pt', \
4 add_special_tokens=False).to('cuda:0')
5output = model.generate(
6 **encoded_input,
7 num_beams=4,
8 do_sample=True,
9 max_new_tokens=100
10)
11
12print(tokenizer.decode(output[0], skip_special_tokens=True))>>> Нейронная сеть — это математическая модель, состоящая из большого
количества нейронов, соединенных между собой электрическими связями.
Нейронная сеть может быть смоделирована на компьютере, и с ее помощью
можно решать задачи, которые не поддаются решению с помощью традиционных
математических методов.1request = "Гагарин полетел в космос в"
2
3encoded_input = tokenizer(request, return_tensors='pt', \
4 add_special_tokens=False).to('cuda:0')
5output = model.generate(
6 **encoded_input,
7 num_beams=2,
8 do_sample=True,
9 max_new_tokens=100
10)
11
12print(tokenizer.decode(output[0], skip_special_tokens=True))>>> Гагарин полетел в космос в 1961 году. Это было первое в истории
человечества космическое путешествие. Юрий Гагарин совершил его
на космическом корабле Восток-1. Корабль был запущен с космодрома
Байконур.