Spaces:
Running
on
Zero
Running
on
Zero
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(
|