final

app.py

@@ -35,10 +35,11 @@ class Steer(pv.SourcelessIntervention):
         if subspaces is None:
             return base
         steering_vec = []
+        avg_mag = sum(subspaces["mag"]) / len(subspaces["mag"])
         for idx, mag in zip(subspaces["idx"], subspaces["mag"]):
             steering_vec.append(self.proj.weight[idx].unsqueeze(dim=0))
         steering_vec = torch.cat(steering_vec, dim=0).mean(dim=0)
-        steering_vec = mag * steering_vec
+        steering_vec = avg_mag * steering_vec
         return base + steering_vec
 
 # Check GPU