pyvene
/

gemma-reft-r1-2b-it-res

Model card Files Files and versions Community

zhengxuanzenwu commited on Feb 13

Commit

eb8b8eb

·

verified ·

1 Parent(s): 5b69dda

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -44,7 +44,7 @@ class Encoder(pv.CollectIntervention):
         return torch.relu(self.proj(base))
 # Loading weights
-path_to_params = hf_hub_download(repo_id="pyvene/gemma-reft-2b-it-res", filename="l20/weight.pt")
 params = torch.load(path_to_params)
 encoder = Encoder(embed_dim=params.shape[0], latent_dim=params.shape[1])
 encoder.proj.weight.data = params.float()

         return torch.relu(self.proj(base))
 # Loading weights
+path_to_params = hf_hub_download(repo_id="pyvene/gemma-reft-r1-2b-it-res", filename="l20/weight.pt")
 params = torch.load(path_to_params)
 encoder = Encoder(embed_dim=params.shape[0], latent_dim=params.shape[1])
 encoder.proj.weight.data = params.float()