Skip to content

Commit 6bcd695

Browse files
committed
test cancelation 2
1 parent 07175a0 commit 6bcd695

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

test/convergence/test_mini_models_multimodal.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,5 @@
11
import functools
22
import os
3-
4-
import pytest
5-
import torch
6-
from datasets import load_dataset
7-
from torch.utils.data import DataLoader
8-
from transformers.models.auto.processing_auto import AutoProcessor
9-
10-
from liger_kernel.transformers import apply_liger_kernel_to_qwen2_vl
113
from test.utils import (
124
UNTOKENIZED_DATASET_PATH,
135
MiniModelConfig,
@@ -18,6 +10,14 @@
1810
supports_bfloat16,
1911
)
2012

13+
import pytest
14+
import torch
15+
from datasets import load_dataset
16+
from torch.utils.data import DataLoader
17+
from transformers.models.auto.processing_auto import AutoProcessor
18+
19+
from liger_kernel.transformers import apply_liger_kernel_to_qwen2_vl
20+
2121
try:
2222
# Qwen2-VL is only available in transformers>4.44.2
2323
from transformers.models.qwen2_vl.configuration_qwen2_vl import Qwen2VLConfig
@@ -305,7 +305,7 @@ def test_mini_model_multimodal(
305305
)
306306

307307
# Compare the params from the last step
308-
# Iterate over the model's parameters and compare thema
308+
# Iterate over the model's parameters and compare them
309309
for expected_param, actual_param in zip(
310310
expected_output["model"].named_parameters(),
311311
actual_output["model"].named_parameters(),

0 commit comments

Comments
 (0)