fix #9

hiyouga · Jul 13, 2023 · f23c7b8 · f23c7b8
1 parent d788f20
commit f23c7b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fastedit/rome/compute_v.py b/fastedit/rome/compute_v.py
@@ -158,7 +158,7 @@ def edit_output_fn(cur_out, cur_layer):
     print(f"Change in target norm: {np.round(target_init.norm().item(), 3)} to {np.round(target.norm().item(), 3)} => "
           f"{np.round((target.norm() - target_init.norm()).item(), 3)}")
     print(f"Division Factor: {np.round(torch.dot(cur_input, left_vector).item(), 3)}")
-    print(f"Right vector norm: {np.round(right_vector.norm(), 3)}")
+    print(f"Right vector norm: {np.round(right_vector.norm().item(), 3)}")
 
     return right_vector