Spaces:
Sleeping
Sleeping
change model
Browse files
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-
|
| 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"
|