shftan commited on
Commit
7193e50
·
1 Parent(s): 0f93598

change model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -66,7 +66,7 @@ def launch_app():
66
  yield output_html
67
 
68
  # Set model, interpreter, dictionary choices
69
- model_name = "google/gemma-3-270m-it" #"google/gemma-2-2b-it"
70
  interpreter_name = "pyvene/gemma-reft-r1-2b-it-res"
71
  interpreter_path = "l20/weight.pt"
72
  interpreter_component = "model.layers[20].output"
 
66
  yield output_html
67
 
68
  # Set model, interpreter, dictionary choices
69
+ model_name = "google/gemma-2-2b-it"
70
  interpreter_name = "pyvene/gemma-reft-r1-2b-it-res"
71
  interpreter_path = "l20/weight.pt"
72
  interpreter_component = "model.layers[20].output"