Spaces:

Amitontheweb
/

Params_Playground

Sleeping

App Files Files Community

Amitontheweb commited on Aug 19

Commit

18abbba

•

1 Parent(s): 0754c2a

Update app.py

Browse files

Files changed (1) hide show

app.py +5 -5

app.py CHANGED Viewed

@@ -36,7 +36,7 @@ def generate(input_text, number_steps, number_beams, number_beam_groups, diversi
         top_p=top_p if top_p_flag else None,
         top_k=top_k if top_k_flag else None,
         no_repeat_ngram_size = no_repeat_ngram_size,
-        repetition_penalty = repetition_penalty if (repetition_penalty > 0) else None,
         output_scores=False,
         do_sample=True
             )
@@ -58,7 +58,7 @@ def generate(input_text, number_steps, number_beams, number_beam_groups, diversi
             length_penalty=length_penalty,
             temperature=temperature if beam_temp_flag else None,
             no_repeat_ngram_size = no_repeat_ngram_size,
-            repetition_penalty = repetition_penalty if (repetition_penalty > 0) else None,
             early_stopping = True if early_stop_flag else False,
             output_scores=False,
             do_sample=True if beam_temp_flag else False
@@ -94,7 +94,7 @@ def generate(input_text, number_steps, number_beams, number_beam_groups, diversi
             return_dict_in_generate=False,
             length_penalty=length_penalty,
             no_repeat_ngram_size = no_repeat_ngram_size,
-            repetition_penalty = repetition_penalty if (repetition_penalty > 0) else None,
             early_stopping = True if early_stop_flag else False,
             output_scores=False,
             )
@@ -117,7 +117,7 @@ def generate(input_text, number_steps, number_beams, number_beam_groups, diversi
         penalty_alpha=penalty_alpha,
         top_k=top_k if top_k_flag else None,
         no_repeat_ngram_size = no_repeat_ngram_size,
-        repetition_penalty = repetition_penalty if (repetition_penalty > 0) else None,
         output_scores=False,
         do_sample=True
             )
@@ -418,7 +418,7 @@ with gr.Blocks() as demo:
             ## Strategies:
-            Given some text as input, a decoder-only model hunts for the most popular continuation - whether the continuation makes sense or not - using various search strategies.
             Example:

         top_p=top_p if top_p_flag else None,
         top_k=top_k if top_k_flag else None,
         no_repeat_ngram_size = no_repeat_ngram_size,
+        repetition_penalty = float(repetition_penalty) if (repetition_penalty > 0) else None,
         output_scores=False,
         do_sample=True
             )
             length_penalty=length_penalty,
             temperature=temperature if beam_temp_flag else None,
             no_repeat_ngram_size = no_repeat_ngram_size,
+            repetition_penalty = float(repetition_penalty) if (repetition_penalty > 0) else None,
             early_stopping = True if early_stop_flag else False,
             output_scores=False,
             do_sample=True if beam_temp_flag else False
             return_dict_in_generate=False,
             length_penalty=length_penalty,
             no_repeat_ngram_size = no_repeat_ngram_size,
+            repetition_penalty = float(repetition_penalty) if (repetition_penalty > 0) else None,
             early_stopping = True if early_stop_flag else False,
             output_scores=False,
             )
         penalty_alpha=penalty_alpha,
         top_k=top_k if top_k_flag else None,
         no_repeat_ngram_size = no_repeat_ngram_size,
+        repetition_penalty = float(repetition_penalty) if (repetition_penalty > 0) else None,
         output_scores=False,
         do_sample=True
             )
             ## Strategies:
+            Given some text as input, a decoder-only model hunts for a continuation using various search strategies. (Whether the continuation makes sense or not is for us to determine.)
             Example: