[CPU] support the case where num_attention_heads or intermediate_size is not divisible by the TP size (#6771)
This commit is contained in:
@@ -13,6 +13,8 @@
|
||||
# ==============================================================================
|
||||
"""Common utilities."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import base64
|
||||
import builtins
|
||||
import ctypes
|
||||
|
||||
Reference in New Issue
Block a user