Update README.md

README.md

@@ -42,7 +42,7 @@ class Encoder(pv.CollectIntervention):
         return torch.relu(self.proj(base))
 
 # Loading weights
-path_to_params = hf_hub_download(repo_id="pyvene/gemma-reft-9b-it-res", filename="l20/weight.pt")
+path_to_params = hf_hub_download(repo_id="pyvene/gemma-reft-r1-9b-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()