Skip to content

Commit 19b4d54

Browse files
authored
Merge branch 'main' into ipadapter-flux
2 parents 45a2fb1 + f66bd32 commit 19b4d54

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/diffusers/models/embeddings.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -691,7 +691,7 @@ def _get_positional_embeddings(
691691
output_type="pt",
692692
)
693693
pos_embedding = pos_embedding.flatten(0, 1)
694-
joint_pos_embedding = torch.zeros(
694+
joint_pos_embedding = pos_embedding.new_zeros(
695695
1, self.max_text_seq_length + num_patches, self.embed_dim, requires_grad=False
696696
)
697697
joint_pos_embedding.data[:, self.max_text_seq_length :].copy_(pos_embedding)

tests/pipelines/mochi/test_mochi.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -275,7 +275,7 @@ def tearDown(self):
275275
gc.collect()
276276
torch.cuda.empty_cache()
277277

278-
def test_cogvideox(self):
278+
def test_mochi(self):
279279
generator = torch.Generator("cpu").manual_seed(0)
280280

281281
pipe = MochiPipeline.from_pretrained("genmo/mochi-1-preview", torch_dtype=torch.float16)

0 commit comments

Comments
 (0)