Skip to content

Commit 99961fc

Browse files
committed
fixes
1 parent cdada86 commit 99961fc

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

src/transformers/models/bark/modeling_bark.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1299,6 +1299,8 @@ def __init__(self, config):
12991299

13001300
self.config = config
13011301

1302+
self.post_init()
1303+
13021304
@classmethod
13031305
def can_generate(cls) -> bool:
13041306
# Bark has a unique model structure, where the external class (`BarkModel`) doesn't need to inherit from

src/transformers/models/distilbert/modeling_distilbert.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -66,9 +66,9 @@ def create_sinusoidal_embeddings(n_pos: int, dim: int, out: torch.Tensor):
6666

6767
with deepspeed.zero.GatheredParameters(out, modifier_rank=0):
6868
if torch.distributed.get_rank() == 0:
69-
_create_sinusoidal_embeddings(n_pos=n_pos, dim=dim, out=out)
69+
return _create_sinusoidal_embeddings(n_pos=n_pos, dim=dim, out=out)
7070
else:
71-
_create_sinusoidal_embeddings(n_pos=n_pos, dim=dim, out=out)
71+
return _create_sinusoidal_embeddings(n_pos=n_pos, dim=dim, out=out)
7272

7373

7474
def _create_sinusoidal_embeddings(n_pos: int, dim: int, out: torch.Tensor):

0 commit comments

Comments
 (0)