fix typo name (microsoft#3689)

Co-authored-by: Logan Adams <[email protected]> Co-authored-by: Michael Wyatt <[email protected]>
rui-ren · Jun 6, 2023 · f2f5f21 · f2f5f21
1 parent d8aaa58
commit f2f5f21
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepspeed/module_inject/containers/features/split_qkv.py b/deepspeed/module_inject/containers/features/split_qkv.py
@@ -123,7 +123,7 @@ def reset_qkv_experimental(self):
         for data in qkv_data:
             del data
 
-    def set_attn_parameters_wo_copy(self, Z3_enabled=False):
+    def set_attn_params_wo_copy(self, Z3_enabled=False):
         self.module.attention.attn_ow = self.dense_w
         self.module.attention.attn_ob = self.dense_b
         if not Z3_enabled: