lukasz-staniszewski commited on
Commit
2cb17f9
·
verified ·
1 Parent(s): e6dd0dc

Refresh model card with renamed concept

Browse files
Files changed (1) hide show
  1. README.md +7 -36
README.md CHANGED
@@ -2,52 +2,23 @@
2
  library_name: audio-interv
3
  tags:
4
  - ace-step
5
- - activation-steering
6
  - audio
7
  - austeer
8
- - diffusion
9
- - genre
10
  - interpretability
11
  - music
12
- - steering
13
  ---
14
 
15
- # Steering Vectors — `genre` (ACE-Step)
16
 
17
- Steering vectors for the **genre** concept on ACE-Step, computed via contrastive activation addition (CAA).
18
 
19
  ## Quickstart
20
 
21
  ```python
22
- from src.steering import SteerableACEModel
23
- from steering.caa import CAAController
24
 
25
- model = SteerableACEModel(device="cuda")
26
- ctrl = CAAController.from_pretrained("lukasz-staniszewski/ace-step-austeer-genre-tf6tf7", alpha=20)
27
-
28
- with model.steer(ctrl):
29
- audio = model.generate(prompt="instrumental music", seed=0)
30
- ```
31
-
32
- ## Generation config
33
-
34
- ```json
35
- {
36
- "method": "austeer",
37
- "concept": "genre",
38
- "lyrics": "[inst]",
39
- "layers": "tf6tf7",
40
- "layers_collected": [
41
- "tf6",
42
- "tf7"
43
- ],
44
- "num_inference_steps": 30,
45
- "audio_duration": 30.0,
46
- "seed": 10,
47
- "guidance_scale": 5.0,
48
- "guidance_scale_text": 0.0,
49
- "guidance_scale_lyric": 0.0,
50
- "guidance_interval": 1.0,
51
- "guidance_interval_decay": 0.0
52
- }
53
  ```
 
2
  library_name: audio-interv
3
  tags:
4
  - ace-step
 
5
  - audio
6
  - austeer
7
+ - electronic_music
 
8
  - interpretability
9
  - music
 
10
  ---
11
 
12
+ # AUSteer Scores — `electronic_music` (layers=tf6tf7) — ACE-Step
13
 
14
+ AUSteer activation-momentum scores for the **electronic_music** concept on ACE-Step. Consumed by `AuSteerController.from_pretrained`. The controller does global top-k feature selection at runtime over the active layer set.
15
 
16
  ## Quickstart
17
 
18
  ```python
19
+ from src.steering import SteerableACEModel, get_method
20
+ import src.steering.methods.austeer
21
 
22
+ model = SteerableACEModel(device="cuda"); model.pipeline.load()
23
+ ctrl = get_method("austeer").from_pretrained("lukasz-staniszewski/ace-step-austeer-electronic-music-tf6tf7", alpha=20)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
24
  ```