Hugging Face's logo

Spaces:
afg1
/
plantRNAFM-saes
Running

App Files Community
afg1 commited on
Commit
4961530
·
verified ·
1 Parent(s): 2f84aa2

Add all the other layers as options

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -221,7 +221,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
221
  )
222
 
223
  with gr.Row():
224
- sae_layer = gr.Dropdown(label="SAE Layer", choices=[3, 6, 9], value=6)
225
  sae_sparsity = gr.Dropdown(label="SAE Sparsity (k)", choices=[32, 64, 128, 256, 512], value=128)
226
 
227
  with gr.Accordion("Input Method", open=True):
 
221
  )
222
 
223
  with gr.Row():
224
+ sae_layer = gr.Dropdown(label="SAE Layer", choices=[1,2,3,4,5,6,7,8,9,10,11], value=6)
225
  sae_sparsity = gr.Dropdown(label="SAE Sparsity (k)", choices=[32, 64, 128, 256, 512], value=128)
226
 
227
  with gr.Accordion("Input Method", open=True):