Update app.py
Browse files
app.py
CHANGED
@@ -51,7 +51,7 @@ your_prompt = gr.Textbox(label="Your Prompt", info="Your Prompt that you wanna m
|
|
51 |
|
52 |
task_prefix = gr.Textbox(label="Task Prefix", info="The prompt prefix for how the AI should make yours better",value="Expand the following prompt to add more detail")
|
53 |
|
54 |
-
max_new_tokens = gr.Slider(value=512, minimum=
|
55 |
|
56 |
repetition_penalty = gr.Slider(value=1.2, minimum=0, maximum=2.0, step=0.05, label="Repetition Penalty", info="Penalize repeated tokens, making the AI repeat less itself")
|
57 |
|
|
|
51 |
|
52 |
task_prefix = gr.Textbox(label="Task Prefix", info="The prompt prefix for how the AI should make yours better",value="Expand the following prompt to add more detail")
|
53 |
|
54 |
+
max_new_tokens = gr.Slider(value=512, minimum=25, maximum=512, step=1, label="Max New Tokens", info="The maximum numbers of new tokens, controls how long is the output")
|
55 |
|
56 |
repetition_penalty = gr.Slider(value=1.2, minimum=0, maximum=2.0, step=0.05, label="Repetition Penalty", info="Penalize repeated tokens, making the AI repeat less itself")
|
57 |
|