cfvg
Browse files
app.py
CHANGED
@@ -54,7 +54,7 @@ def generate_paraphrases(text, setting, output_format):
|
|
54 |
num_return_sequences = 5
|
55 |
repetition_penalty = 1.1
|
56 |
no_repeat_ngram_size = 2
|
57 |
-
temperature =
|
58 |
max_length = 128
|
59 |
elif setting == 2:
|
60 |
num_return_sequences = 10
|
@@ -81,8 +81,8 @@ def generate_paraphrases(text, setting, output_format):
|
|
81 |
temperature = 1.8
|
82 |
max_length = 384
|
83 |
|
84 |
-
top_k =
|
85 |
-
top_p = 0.
|
86 |
length_penalty = 1.0
|
87 |
|
88 |
formatted_output = "Original text:\n" + text + "\n\n"
|
|
|
54 |
num_return_sequences = 5
|
55 |
repetition_penalty = 1.1
|
56 |
no_repeat_ngram_size = 2
|
57 |
+
temperature = 0.8
|
58 |
max_length = 128
|
59 |
elif setting == 2:
|
60 |
num_return_sequences = 10
|
|
|
81 |
temperature = 1.8
|
82 |
max_length = 384
|
83 |
|
84 |
+
top_k = 40
|
85 |
+
top_p = 0.90
|
86 |
length_penalty = 1.0
|
87 |
|
88 |
formatted_output = "Original text:\n" + text + "\n\n"
|