frankaging commited on
Commit
b870c53
·
1 Parent(s): b58a148
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -245,7 +245,7 @@ with gr.Blocks(css=css, fill_height=True) as demo:
245
  chat_interface = gr.ChatInterface(
246
  fn=generate,
247
  title="Chat with a Concept Steering Model",
248
- description="""Steer responses by selecting concepts on the right →\n\nWe are using Gemma-2-2B-it with steering vectors added to the residual stream at layer 20. Our auto-steer steering vector generated is a finetuned Gemma-2-2B model. Steered model tends to generate longer responses.""",
249
  type="messages",
250
  additional_inputs=[selected_subspaces],
251
  fill_height=True
@@ -254,7 +254,7 @@ with gr.Blocks(css=css, fill_height=True) as demo:
254
  # Right side: concept management
255
  with gr.Column(scale=3):
256
  gr.Markdown("## Steer Model Responses")
257
- gr.Markdown("Search and then select a concept to steer. The closest match will be automatically selected. If there is no match, we will use our steering vector generator to auto-steer for you!")
258
  # Concept Search and Selection
259
  with gr.Group():
260
  search_box = gr.Textbox(
 
245
  chat_interface = gr.ChatInterface(
246
  fn=generate,
247
  title="Chat with a Concept Steering Model",
248
+ description="""Steer responses by selecting concepts on the right →\n\nWe intervene on Gemma-2-2B-it by adding steering vectors to the residual stream at layer 20.""",
249
  type="messages",
250
  additional_inputs=[selected_subspaces],
251
  fill_height=True
 
254
  # Right side: concept management
255
  with gr.Column(scale=3):
256
  gr.Markdown("## Steer Model Responses")
257
+ gr.Markdown("Search and then select a concept to steer. The closest match will be automatically selected. If there is no match, our steering vector generator (a finetuned Gemma-2-2B) auto-steers for you!")
258
  # Concept Search and Selection
259
  with gr.Group():
260
  search_box = gr.Textbox(