Spaces:
Runtime error
Runtime error
frankaging
commited on
Commit
·
b870c53
1
Parent(s):
b58a148
final?
Browse files
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
|
| 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,
|
| 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(
|