Spaces:
Runtime error
Runtime error
Commit
·
b01cd50
1
Parent(s):
33ecc84
Update app.py
Browse files
app.py
CHANGED
@@ -4,13 +4,13 @@ st.header("Transformer parameters")
|
|
4 |
col1, col2 = st.columns([2, 4])
|
5 |
|
6 |
col1.write('Batch size: ')
|
7 |
-
bs = col2.number_input('
|
8 |
col1.write('Num heads:')
|
9 |
-
h = col2.number_input('
|
10 |
col1.write('Dimension:')
|
11 |
-
d = col2.number_input('
|
12 |
col1.write('Seq length:')
|
13 |
-
n = col2.number_input('
|
14 |
|
15 |
st.header('Query, Key, Value projection')
|
16 |
|
|
|
4 |
col1, col2 = st.columns([2, 4])
|
5 |
|
6 |
col1.write('Batch size: ')
|
7 |
+
bs = col2.number_input('', value=10)
|
8 |
col1.write('Num heads:')
|
9 |
+
h = col2.number_input('', value=16)
|
10 |
col1.write('Dimension:')
|
11 |
+
d = col2.number_input('', value=768)
|
12 |
col1.write('Seq length:')
|
13 |
+
n = col2.number_input('', value=1024)
|
14 |
|
15 |
st.header('Query, Key, Value projection')
|
16 |
|