Updated hyperparameters
Browse files
app.py
CHANGED
@@ -752,7 +752,7 @@ stepsize=2.56
|
|
752 |
temperature=1.0
|
753 |
top_k=2
|
754 |
sample=True
|
755 |
-
num_iterations=
|
756 |
grad_length=10000
|
757 |
horizon_length=1
|
758 |
window_length=0
|
@@ -813,7 +813,7 @@ for param in model.parameters():
|
|
813 |
|
814 |
eot_token = "<|endoftext|>"
|
815 |
|
816 |
-
def get_reply(response, history = None, in_stepsize = 2.56, in_horizon_length = 1, in_num_iterations =
|
817 |
stepsize = in_stepsize
|
818 |
horizon_length = int(in_horizon_length)
|
819 |
num_iterations = int(in_num_iterations)
|
@@ -888,11 +888,6 @@ css = """
|
|
888 |
|
889 |
gr.Interface(fn=get_reply,
|
890 |
theme="default",
|
891 |
-
inputs=[gr.inputs.Textbox(placeholder="How are you?"),
|
892 |
-
"state",
|
893 |
-
gr.inputs.Number(default=2.56, label="Step"),
|
894 |
-
gr.inputs.Number(default=1, label="Horizon"),
|
895 |
-
gr.inputs.Number(default=10, label="Iterations"),
|
896 |
-
gr.inputs.Number(default=2, label="Top_k")],
|
897 |
outputs=["html", "state"],
|
898 |
css=css).launch()
|
|
|
752 |
temperature=1.0
|
753 |
top_k=2
|
754 |
sample=True
|
755 |
+
num_iterations=0
|
756 |
grad_length=10000
|
757 |
horizon_length=1
|
758 |
window_length=0
|
|
|
813 |
|
814 |
eot_token = "<|endoftext|>"
|
815 |
|
816 |
+
def get_reply(response, history = None, in_stepsize = 2.56, in_horizon_length = 1, in_num_iterations = 0, in_top_k = 2):
|
817 |
stepsize = in_stepsize
|
818 |
horizon_length = int(in_horizon_length)
|
819 |
num_iterations = int(in_num_iterations)
|
|
|
888 |
|
889 |
gr.Interface(fn=get_reply,
|
890 |
theme="default",
|
891 |
+
inputs=[gr.inputs.Textbox(placeholder="How are you?"), "state"],
|
|
|
|
|
|
|
|
|
|
|
892 |
outputs=["html", "state"],
|
893 |
css=css).launch()
|