Skip to content

Commit 3a1a195

Browse files
fix
1 parent 82d1cf9 commit 3a1a195

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

tests/fx/parallelization/test_tensor_parallel.py

+1-5
Original file line numberDiff line numberDiff line change
@@ -189,11 +189,7 @@ def run_test_tie_word_embeddings(rank: int, world_size: int, model_id: str, mode
189189
if isinstance(mod, ColumnParallelLinear):
190190
lm_head_weight = mod.weight
191191
break
192-
assert (
193-
id(embedding_weight) == id(lm_head_weight)
194-
and hasattr(embedding_weight, "meta")
195-
and embedding_weight.meta.is_tied
196-
)
192+
assert id(embedding_weight) == id(lm_head_weight)
197193
dist.barrier(tp_group)
198194
tearDown()
199195

0 commit comments

Comments
 (0)