Harvard-DCML
/

boomerang-pythia-1.6B

Model card Files Files and versions

s-kangasl commited on 8 days ago

Commit

643e51f

·

verified ·

1 Parent(s): b4c8937

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -36,7 +36,7 @@ from patching.patch import build_intermediate_model
 intermediate_model = build_intermediate_model(
   teacher_name_or_path = "EleutherAI/pythia-2.8b",
-  student_name_or_path = "s-kangasl/boomerang-pythia-1.6B",
   num_layers_to_patch = 2,
   patch_first_k_layers = False,
   dtype = torch.bfloat16,

 intermediate_model = build_intermediate_model(
   teacher_name_or_path = "EleutherAI/pythia-2.8b",
+  student_name_or_path = "Harvard-DCML/boomerang-pythia-1.6B",
   num_layers_to_patch = 2,
   patch_first_k_layers = False,
   dtype = torch.bfloat16,