Spaces:

pyvene
/

AxBench-ReFT-r1-16K

Running on Zero

frankaging commited on Jan 25

Commit

17a04d5

1 Parent(s): 884bfb5

o1 impl

Files changed (1) hide show

app.py CHANGED Viewed

@@ -67,7 +67,7 @@ if torch.cuda.is_available():
     steer = Steer(embed_dim=params.shape[0], latent_dim=params.shape[1])
     steer.proj.weight.data = params.float()
-    pv_model = pyreft.IntervenableModel(
         {
             "component": f"model.layers[20].output",
             "intervention": steer,

     steer = Steer(embed_dim=params.shape[0], latent_dim=params.shape[1])
     steer.proj.weight.data = params.float()
+    pv_model = pv.IntervenableModel(
         {
             "component": f"model.layers[20].output",
             "intervention": steer,