Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
Co-authored-by: Michael Wyatt <michaelwyatt@microsoft.com>
This commit is contained in:
tensor-tang 2023-06-07 07:43:29 +08:00 коммит произвёл GitHub
Родитель d8aaa58122
Коммит f2f5f21b52
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -123,7 +123,7 @@ class HybridSplitQKVContainer(HybridEngineContainer):
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: