Spaces:
Running
Running
Update README.md
Browse files
README.md
CHANGED
|
@@ -22,22 +22,6 @@ Each model repository includes:
|
|
| 22 |
|
| 23 |
## Quick Start
|
| 24 |
|
| 25 |
-
```python
|
| 26 |
-
from open_concept_steering import ModelLoader, SAETrainer, FeatureScanner, ChatInterface
|
| 27 |
-
|
| 28 |
-
# Load a model and train SAE
|
| 29 |
-
model = ModelLoader.load("llama2")
|
| 30 |
-
sae = SAETrainer.train(model)
|
| 31 |
-
|
| 32 |
-
# Find features for a concept
|
| 33 |
-
scanner = FeatureScanner(sae)
|
| 34 |
-
feature = scanner.find_feature("golden gate bridge")
|
| 35 |
-
|
| 36 |
-
# Chat with the steered model
|
| 37 |
-
chat = ChatInterface(model)
|
| 38 |
-
chat.run(amplified_features=[feature])
|
| 39 |
-
```
|
| 40 |
-
|
| 41 |
## Examples
|
| 42 |
|
| 43 |
See the `examples/` directory for detailed notebooks demonstrating:
|
|
|
|
| 22 |
|
| 23 |
## Quick Start
|
| 24 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 25 |
## Examples
|
| 26 |
|
| 27 |
See the `examples/` directory for detailed notebooks demonstrating:
|