riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

fix: replace asserts by error (#13894)

Browse Source
This commit is contained in:
Siarhei Melnik 2021-10-06 01:08:48 +03:00 committed by GitHub
parent f099249cf1
commit 7af7d7ce05
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/transformers/models/distilbert/modeling_flax_distilbert.py
View File

@ -172,7 +172,8 @@ class FlaxMultiHeadSelfAttention(nn.Module):
self.dim = self.config.dim
self.dropout = nn.Dropout(rate=self.config.attention_dropout)
assert self.dim % self.n_heads == 0, f"Hidden size {self.dim} not dividable by number of heads {self.n_heads}"
if not (self.dim % self.n_heads == 0):
raise ValueError(f"Hidden size {self.dim} not dividable by number of heads {self.n_heads}")
self.q_lin = nn.Dense(
self.dim,