lukasz-staniszewski commited on
Commit
15a8465
·
verified ·
1 Parent(s): 8155160

Fix model card: use AudioLDMCAASteeringController + AudioLDM2 backbone

Browse files
Files changed (1) hide show
  1. README.md +10 -8
README.md CHANGED
@@ -1,7 +1,6 @@
1
  ---
2
  library_name: audio-interv
3
  tags:
4
- - ace-step
5
  - activation-steering
6
  - audio
7
  - audioldm
@@ -14,21 +13,24 @@ tags:
14
  - steering
15
  ---
16
 
17
- # Steering Vectors — `female_vocals` (ACE-Step)
18
 
19
- Steering vectors for the **female_vocals** concept on ACE-Step, computed via contrastive activation addition (CAA).
20
 
21
  ## Quickstart
22
 
23
  ```python
24
- from src.steering import SteerableACEModel
25
- from src.steering.methods.caa import CAASteeringController
26
 
27
- model = SteerableACEModel(device="cuda")
28
- ctrl = CAASteeringController.from_pretrained("lukasz-staniszewski/audioldm2-caa-female-vocals", alpha=20)
29
 
30
  with model.steer(ctrl):
31
- audio = model.generate(prompt="instrumental music", seed=0)
 
 
 
 
32
  ```
33
 
34
  ## Generation config
 
1
  ---
2
  library_name: audio-interv
3
  tags:
 
4
  - activation-steering
5
  - audio
6
  - audioldm
 
13
  - steering
14
  ---
15
 
16
+ # Steering Vectors — `female_vocals` (AudioLDM2)
17
 
18
+ Steering vectors for the **female_vocals** concept on AudioLDM2, computed via contrastive activation addition (CAA).
19
 
20
  ## Quickstart
21
 
22
  ```python
23
+ from src.steering import SteerableAudioLDMModel, AudioLDMCAASteeringController
 
24
 
25
+ model = SteerableAudioLDMModel(device="cuda")
26
+ ctrl = AudioLDMCAASteeringController.from_pretrained("lukasz-staniszewski/audioldm2-caa-female-vocals", alpha=1.0)
27
 
28
  with model.steer(ctrl):
29
+ out = model.generate(
30
+ prompt="instrumental music",
31
+ num_inference_steps=30, audio_length_in_s=10.0,
32
+ guidance_scale=3.5, seed=0,
33
+ )
34
  ```
35
 
36
  ## Generation config