lukasz-staniszewski commited on
Commit
877592d
·
verified ·
1 Parent(s): f320676

Fix card: AUSteerSteeringController quickstart

Browse files
Files changed (1) hide show
  1. README.md +12 -7
README.md CHANGED
@@ -1,8 +1,8 @@
1
  ---
2
  library_name: audio-interv
3
  tags:
4
- - ace-step
5
  - activation-steering
 
6
  - audio
7
  - austeer
8
  - diffusion
@@ -12,21 +12,26 @@ tags:
12
  - tempo
13
  ---
14
 
15
- # Steering Vectors — `tempo` (ACE-Step)
16
 
17
- Steering vectors for the **tempo** 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-tempo-all", alpha=20)
 
 
 
27
 
28
  with model.steer(ctrl):
29
- audio = model.generate(prompt="instrumental music", seed=0)
 
 
 
30
  ```
31
 
32
  ## Generation config
 
1
  ---
2
  library_name: audio-interv
3
  tags:
 
4
  - activation-steering
5
+ - all
6
  - audio
7
  - austeer
8
  - diffusion
 
12
  - tempo
13
  ---
14
 
15
+ # AUSteer — `tempo` (ACE-Step)
16
 
17
+ Per-(step, layer) sparse activation-momentum scores for the **tempo** 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-tempo-all", 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