Ilvir commited on
Commit
9a633bf
·
1 Parent(s): 6af1775

Update gpt.py

Browse files
Files changed (1) hide show
  1. gpt.py +2 -2
gpt.py CHANGED
@@ -47,13 +47,13 @@ if st.button('Сгенерировать текст'):
47
  out = model.generate(
48
  input_ids=prompt,
49
  max_length=length,
50
- num_beams=num_samples,
51
  do_sample=True,
52
  temperature=temperature,
53
  top_k=top_k,
54
  top_p=top_p,
55
  no_repeat_ngram_size=3,
56
- num_return_sequences=3,
57
  ).cpu().numpy()
58
  for i, out_ in enumerate(out):
59
  st.write(f'Текст {i+1}:')
 
47
  out = model.generate(
48
  input_ids=prompt,
49
  max_length=length,
50
+ num_beams=5,
51
  do_sample=True,
52
  temperature=temperature,
53
  top_k=top_k,
54
  top_p=top_p,
55
  no_repeat_ngram_size=3,
56
+ num_return_sequences=num_samples,
57
  ).cpu().numpy()
58
  for i, out_ in enumerate(out):
59
  st.write(f'Текст {i+1}:')