Update src/app.py
Browse files- src/app.py +3 -3
src/app.py
CHANGED
|
@@ -118,10 +118,10 @@ with col.expander("Detailed Information", expanded=False):
|
|
| 118 |
st.markdown("""- For LoRa Fine-tuning, This assumes a **16-bit** dtype of trainable parameters. The formula (in terms of GB) is""")
|
| 119 |
st.latex(r"\text{Memory}_\text{LoRa} \approx \text{Model Size} + \left(\text{ \# trainable Params}_\text{Billions}\times\frac{16}{8} \times 4\right) \times 1.2")
|
| 120 |
|
| 121 |
-
|
| 122 |
|
| 123 |
-
|
| 124 |
-
|
| 125 |
|
| 126 |
#model_name = st.sidebar.text_input("Model name", value="mistralai/Mistral-7B-v0.1")
|
| 127 |
with st.sidebar.container():
|
|
|
|
| 118 |
st.markdown("""- For LoRa Fine-tuning, This assumes a **16-bit** dtype of trainable parameters. The formula (in terms of GB) is""")
|
| 119 |
st.latex(r"\text{Memory}_\text{LoRa} \approx \text{Model Size} + \left(\text{ \# trainable Params}_\text{Billions}\times\frac{16}{8} \times 4\right) \times 1.2")
|
| 120 |
|
| 121 |
+
access_token = st.sidebar.text_input("Access token")
|
| 122 |
|
| 123 |
+
if access_token:
|
| 124 |
+
login(token=access_token)
|
| 125 |
|
| 126 |
#model_name = st.sidebar.text_input("Model name", value="mistralai/Mistral-7B-v0.1")
|
| 127 |
with st.sidebar.container():
|