Kirill U-Boot commited on
Commit
017a3cb
·
1 Parent(s): 0473d1d

Изменение в README.md (#1)

Browse files

- Изменение в README.md (91fbe933ddf46103198663f0bc28c7e4687e4966)


Co-authored-by: U-Boot <[email protected]>

Files changed (1) hide show
  1. README.md +18 -0
README.md CHANGED
@@ -1,3 +1,21 @@
1
  ---
2
  license: mit
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
  ---
4
+ ## Привет! Это модель основанная на ruGPT3-small
5
+ Модель опубликована под лицензией MIT!
6
+ # Пример кода на Python
7
+ ```python
8
+ from transformers import GPT2LMHeadModel, GPT2Tokenizer
9
+
10
+ model_name_or_path = "путь до папки с моделью"
11
+ tokenizer = GPT2Tokenizer.from_pretrained(model_name_or_path)
12
+ model = GPT2LMHeadModel.from_pretrained(model_name_or_path).cuda()
13
+
14
+ input_ids = tokenizer.encode(message.content, return_tensors="pt").cuda()
15
+ out = model.generate(input_ids.cuda(), repetition_penalty=5.0, do_sample=True, top_k=5, top_p=0.95, temperature=1)
16
+ generated_text = list(map(tokenizer.decode, out))
17
+
18
+ print(generated_text[0])
19
+ ```
20
+ # Смешные текста
21
+ Пока их нет! Но если ты хочешь добавить смешной случай открывай pull request