Spaces:

hbfreed
/

olmo2-sae-steering-demo

Running on Zero

hbfreed commited on Jun 3

Commit

362bcd0

verified ·

1 Parent(s): a28f630

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -48,9 +48,9 @@ model.set_sae_and_layer(sae, steering_layer)
 # Steering features configuration
 STEERING_FEATURES = {
     "None": {"feature": None, "default": 0, "name": "No Steering"},
-    "batman/bruce wayne": {"feature": 758, "default": 11, "name": "🦸 Superhero/Batman"},
-    "japan": {"feature": 29940, "default": 13, "name": "🗾 Japan"},
-    "baseball": {"feature": 65023, "default": 6, "name": "⚾ Baseball"}
 }
 default_system_prompt = "You are OLMo 2, a helpful and harmless AI Assistant built by the Allen Institute for AI."

 # Steering features configuration
 STEERING_FEATURES = {
     "None": {"feature": None, "default": 0, "name": "No Steering"},
+    "Batman / Bruce Wayne": {"feature": 758, "default": 9, "name": "🦸 Superhero/Batman"},
+    "Japan": {"feature": 29940, "default": 13, "name": "🗾 Japan"},
+    "Baseball": {"feature": 65023, "default": 6, "name": "⚾ Baseball"}
 }
 default_system_prompt = "You are OLMo 2, a helpful and harmless AI Assistant built by the Allen Institute for AI."