Spaces:
Runtime error
Runtime error
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')
|