Hennara commited on
Commit
6cf8a8a
1 Parent(s): 166583b

Update app.py

Browse files

change the max value of batch

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ def main():
9
  precession = st.sidebar.number_input("precession in Byte", min_value=1, max_value=4, value=2, step=2)
10
  hidden_size = st.sidebar.number_input("Hidden Size", min_value=512, max_value=2 ** 16, value=4096, step=512)
11
  num_heads = st.sidebar.number_input("Number of Attention Heads", min_value=4, max_value=128, value=32, step=4)
12
- batch_size = st.sidebar.number_input("Batch Size", min_value=1, max_value=256, value=64, step=4)
13
  seq_len = st.sidebar.number_input("Sequence Length", min_value=512, max_value=128000, value=2048, step=512)
14
  intermediate_size = st.sidebar.number_input("Intermediate Size", min_value=1024, max_value=2 ** 18, value=11008,
15
  step=128)
 
9
  precession = st.sidebar.number_input("precession in Byte", min_value=1, max_value=4, value=2, step=2)
10
  hidden_size = st.sidebar.number_input("Hidden Size", min_value=512, max_value=2 ** 16, value=4096, step=512)
11
  num_heads = st.sidebar.number_input("Number of Attention Heads", min_value=4, max_value=128, value=32, step=4)
12
+ batch_size = st.sidebar.number_input("Batch Size", min_value=1, max_value=1024, value=64, step=4)
13
  seq_len = st.sidebar.number_input("Sequence Length", min_value=512, max_value=128000, value=2048, step=512)
14
  intermediate_size = st.sidebar.number_input("Intermediate Size", min_value=1024, max_value=2 ** 18, value=11008,
15
  step=128)