Spaces:
Sleeping
Sleeping
Commit
·
b804250
1
Parent(s):
200ef1d
replaced temperature values
Browse files
app.py
CHANGED
@@ -297,7 +297,7 @@ def GenerateText(emo,gpt_tokenizer,gpt_model):
|
|
297 |
generated = gpt_tokenizer(prompt, return_tensors="pt").input_ids
|
298 |
|
299 |
sample_outputs = gpt_model.generate(generated, do_sample=True, top_k=50,
|
300 |
-
max_length=30, top_p=0.95, temperature=1.
|
301 |
|
302 |
# Extract and split the generated text into words
|
303 |
outputs = set([gpt_tokenizer.decode(sample_output, skip_special_tokens=True).split(':')[-1] for sample_output in sample_outputs])
|
|
|
297 |
generated = gpt_tokenizer(prompt, return_tensors="pt").input_ids
|
298 |
|
299 |
sample_outputs = gpt_model.generate(generated, do_sample=True, top_k=50,
|
300 |
+
max_length=30, top_p=0.95, temperature=1.1, num_return_sequences=10)#,no_repeat_ngram_size=1)
|
301 |
|
302 |
# Extract and split the generated text into words
|
303 |
outputs = set([gpt_tokenizer.decode(sample_output, skip_special_tokens=True).split(':')[-1] for sample_output in sample_outputs])
|