lukasz-staniszewski commited on
Commit
55c0b6c
·
verified ·
1 Parent(s): c06d2c3

Upload CAA steering vectors

Browse files
Files changed (5) hide show
  1. README.md +51 -0
  2. config.json +1 -0
  3. neg_vectors.pkl +3 -0
  4. pos_vectors.pkl +3 -0
  5. sv.pkl +3 -0
README.md ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: audio-interv
3
+ tags:
4
+ - ace-step
5
+ - activation-steering
6
+ - audio
7
+ - caa
8
+ - diffusion
9
+ - interpretability
10
+ - mood
11
+ - music
12
+ - steering
13
+ ---
14
+
15
+ # Steering Vectors — `mood` (ACE-Step)
16
+
17
+ Steering vectors for the **mood** 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-caa-mood", 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
+ "concept": "mood",
37
+ "lyrics": "[inst]",
38
+ "num_cfg_passes": 2,
39
+ "save_all_cfg_passes": true,
40
+ "audio_duration": 30.0,
41
+ "num_inference_steps": 30,
42
+ "seed": 10,
43
+ "device": "cuda",
44
+ "save_dir": "steering_vectors",
45
+ "guidance_scale_text": 0.0,
46
+ "guidance_scale_lyric": 0.0,
47
+ "guidance_scale": 5.0,
48
+ "guidance_interval": 1.0,
49
+ "guidance_interval_decay": 0.0
50
+ }
51
+ ```
config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"concept": "mood", "lyrics": "[inst]", "num_cfg_passes": 2, "save_all_cfg_passes": true, "audio_duration": 30.0, "num_inference_steps": 30, "seed": 10, "device": "cuda", "save_dir": "steering_vectors", "guidance_scale_text": 0.0, "guidance_scale_lyric": 0.0, "guidance_scale": 5.0, "guidance_interval": 1.0, "guidance_interval_decay": 0.0}
neg_vectors.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:73f516be00de83ee2f519935fddb63166b7f5099cd5c6743ca06345b8c61ac0a
3
+ size 740442894
pos_vectors.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ead50a843822147d7d0f70729ba8da6376461ab72af2cec9cde1d4216b7da642
3
+ size 740442894
sv.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e702282489bd09bd44fd20aa87eb248b8cbccd5040e173c2c556e0e5d0420f0c
3
+ size 14804843