jeydipak commited on
Commit
c9d6cdd
1 Parent(s): 7d874a1
Files changed (1) hide show
  1. pages/task3.py +2 -2
pages/task3.py CHANGED
@@ -12,8 +12,8 @@ def generate_text(model, tokenizer, prompt, max_length, num_generations, tempera
12
  input_ids,
13
  max_length=max_length,
14
  temperature=temperature,
15
- num_return_sequences=1
16
- )
17
  generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
18
  generated_texts.append(generated_text)
19
 
 
12
  input_ids,
13
  max_length=max_length,
14
  temperature=temperature,
15
+ num_return_sequences=1,
16
+ do_sample=True)
17
  generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
18
  generated_texts.append(generated_text)
19