Spaces:
Running
Running
nicholasKluge
commited on
Commit
•
2677645
1
Parent(s):
53ebaa0
Update app.py
Browse files
app.py
CHANGED
@@ -66,11 +66,11 @@ with gr.Blocks(theme='freddyaboulton/dracula_revamped') as demo:
|
|
66 |
|
67 |
with gr.Accordion(label="Parameters ⚙️", open=True):
|
68 |
safety = gr.Radio(["On", "Off"], label="Guard Rail 🛡️", value="On", info="Helps prevent the model from generating toxic/harmful content.")
|
69 |
-
top_k = gr.Slider(minimum=10, maximum=100, value=
|
70 |
-
top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.
|
71 |
temperature = gr.Slider(minimum=0.1, maximum=2.0, value=0.1, step=0.1, interactive=True, label="Temperature", info="Controls the randomness of the generated tokens.")
|
72 |
-
repetition_penalty = gr.Slider(minimum=1, maximum=2, value=1.
|
73 |
-
max_length = gr.Slider(minimum=10, maximum=500, value=
|
74 |
smaple_from = gr.Slider(minimum=2, maximum=10, value=2, step=1, interactive=True, label="Sample From", info="Controls the number of generations that the reward model will sample from.")
|
75 |
|
76 |
clear = gr.Button("Clear Conversation 🧹")
|
|
|
66 |
|
67 |
with gr.Accordion(label="Parameters ⚙️", open=True):
|
68 |
safety = gr.Radio(["On", "Off"], label="Guard Rail 🛡️", value="On", info="Helps prevent the model from generating toxic/harmful content.")
|
69 |
+
top_k = gr.Slider(minimum=10, maximum=100, value=30, step=5, interactive=True, label="Top-k", info="Controls the number of highest probability tokens to consider for each step.")
|
70 |
+
top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.30, step=0.05, interactive=True, label="Top-p", info="Controls the cumulative probability of the generated tokens.")
|
71 |
temperature = gr.Slider(minimum=0.1, maximum=2.0, value=0.1, step=0.1, interactive=True, label="Temperature", info="Controls the randomness of the generated tokens.")
|
72 |
+
repetition_penalty = gr.Slider(minimum=1, maximum=2, value=1.1, step=0.1, interactive=True, label="Repetition Penalty", info="Higher values help the model to avoid repetition in text generation.")
|
73 |
+
max_length = gr.Slider(minimum=10, maximum=500, value=200, step=10, interactive=True, label="Max Length", info="Controls the maximum length of the generated text.")
|
74 |
smaple_from = gr.Slider(minimum=2, maximum=10, value=2, step=1, interactive=True, label="Sample From", info="Controls the number of generations that the reward model will sample from.")
|
75 |
|
76 |
clear = gr.Button("Clear Conversation 🧹")
|