Spaces:
Running
Running
Vokturz
commited on
Commit
•
2d9aa2d
1
Parent(s):
a0b9dac
fix a warning
Browse files- src/app.py +1 -1
src/app.py
CHANGED
|
@@ -90,7 +90,7 @@ gpu_spec.name = 'INFO'
|
|
| 90 |
|
| 91 |
lora_pct = st.sidebar.slider("LoRa % trainable parameters", 0.1, 100.0, 2.0, step=0.1)
|
| 92 |
|
| 93 |
-
st.sidebar.dataframe(gpu_spec.T)
|
| 94 |
|
| 95 |
memory_table = pd.DataFrame(st.session_state[model_name]).set_index('dtype')
|
| 96 |
memory_table['LoRA Fine-Tuning (GB)'] = (memory_table["Total Size (GB)"] +
|
|
|
|
| 90 |
|
| 91 |
lora_pct = st.sidebar.slider("LoRa % trainable parameters", 0.1, 100.0, 2.0, step=0.1)
|
| 92 |
|
| 93 |
+
st.sidebar.dataframe(gpu_spec.T.astype(str))
|
| 94 |
|
| 95 |
memory_table = pd.DataFrame(st.session_state[model_name]).set_index('dtype')
|
| 96 |
memory_table['LoRA Fine-Tuning (GB)'] = (memory_table["Total Size (GB)"] +
|