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

Fix card: AUSteerSteeringController quickstart

Browse files
Files changed (1) hide show
  1. README.md +42 -8
README.md CHANGED
@@ -1,24 +1,58 @@
1
  ---
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
  ```
 
1
  ---
2
  library_name: audio-interv
3
  tags:
4
+ - activation-steering
5
  - audio
6
  - austeer
7
+ - diffusion
8
+ - genre
9
  - interpretability
10
  - music
11
+ - steering
12
+ - tf6tf7
13
  ---
14
 
15
+ # AUSteer — `genre` (ACE-Step)
16
 
17
+ Per-(step, layer) sparse activation-momentum scores for the **genre** concept on ACE-Step. At inference, `AUSteerSteeringController` adds `alpha` along the top-`k` most concept-discriminative bins.
18
 
19
  ## Quickstart
20
 
21
  ```python
22
+ from src.steering import SteerableACEModel, AUSteerSteeringController
 
23
 
24
+ model = SteerableACEModel(device="cuda")
25
+ model.pipeline.load()
26
+ ctrl = AUSteerSteeringController.from_pretrained(
27
+ "lukasz-staniszewski/ace-step-austeer-electronic-music-tf6tf7", alpha=15.0, k=256, mode="additive",
28
+ )
29
+
30
+ with model.steer(ctrl):
31
+ audio = model.generate(
32
+ prompt="instrumental music", lyrics="[inst]",
33
+ audio_duration=10.0, infer_step=30, manual_seed=0,
34
+ )
35
+ ```
36
+
37
+ ## Generation config
38
+
39
+ ```json
40
+ {
41
+ "method": "austeer",
42
+ "concept": "genre",
43
+ "lyrics": "[inst]",
44
+ "layers": "tf6tf7",
45
+ "layers_collected": [
46
+ "tf6",
47
+ "tf7"
48
+ ],
49
+ "num_inference_steps": 30,
50
+ "audio_duration": 30.0,
51
+ "seed": 10,
52
+ "guidance_scale": 5.0,
53
+ "guidance_scale_text": 0.0,
54
+ "guidance_scale_lyric": 0.0,
55
+ "guidance_interval": 1.0,
56
+ "guidance_interval_decay": 0.0
57
+ }
58
  ```