File tree Expand file tree Collapse file tree 1 file changed +9
-9
lines changed Expand file tree Collapse file tree 1 file changed +9
-9
lines changed Original file line number Diff line number Diff line change 1
1
import functools
2
2
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
11
3
from test .utils import (
12
4
UNTOKENIZED_DATASET_PATH ,
13
5
MiniModelConfig ,
18
10
supports_bfloat16 ,
19
11
)
20
12
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
+
21
21
try :
22
22
# Qwen2-VL is only available in transformers>4.44.2
23
23
from transformers .models .qwen2_vl .configuration_qwen2_vl import Qwen2VLConfig
@@ -305,7 +305,7 @@ def test_mini_model_multimodal(
305
305
)
306
306
307
307
# 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
309
309
for expected_param , actual_param in zip (
310
310
expected_output ["model" ].named_parameters (),
311
311
actual_output ["model" ].named_parameters (),
You can’t perform that action at this time.
0 commit comments