Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ from ast import literal_eval
|
|
7 |
def infer(prompt,
|
8 |
model_name,
|
9 |
max_new_tokens=10,
|
10 |
-
temperature=0.
|
11 |
top_p=1.0,
|
12 |
num_completions=1,
|
13 |
seed=42,
|
@@ -36,7 +36,7 @@ col1, col2 = st.columns([1, 3])
|
|
36 |
with col1:
|
37 |
model_name = st.selectbox("Model", ["GPT-JT-6B-v1"])
|
38 |
max_new_tokens = st.text_input('Max new tokens', "10")
|
39 |
-
temperature = st.text_input('temperature', "0.
|
40 |
top_p = st.text_input('top_p', "1.0")
|
41 |
num_completions = st.text_input('num_completions (only the best one will be returend)', "1")
|
42 |
stop = st.text_input('stop, split by;', r'\n')
|
|
|
7 |
def infer(prompt,
|
8 |
model_name,
|
9 |
max_new_tokens=10,
|
10 |
+
temperature=0.8,
|
11 |
top_p=1.0,
|
12 |
num_completions=1,
|
13 |
seed=42,
|
|
|
36 |
with col1:
|
37 |
model_name = st.selectbox("Model", ["GPT-JT-6B-v1"])
|
38 |
max_new_tokens = st.text_input('Max new tokens', "10")
|
39 |
+
temperature = st.text_input('temperature', "0.8")
|
40 |
top_p = st.text_input('top_p', "1.0")
|
41 |
num_completions = st.text_input('num_completions (only the best one will be returend)', "1")
|
42 |
stop = st.text_input('stop, split by;', r'\n')
|