lambdabrendan commited on
Commit
bfb7105
·
verified ·
1 Parent(s): e1b6a19

Update src/app.py

Browse files
Files changed (1) hide show
  1. 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
- # 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():
 
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():