liger-kernel-nightly 0.5.10.dev20250611191801__tar.gz → 0.5.10.dev20250611215839__tar.gz

{liger_kernel_nightly-0.5.10.dev20250611191801 → liger_kernel_nightly-0.5.10.dev20250611215839}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: liger_kernel_nightly
-Version: 0.5.10.dev20250611191801
+Version: 0.5.10.dev20250611215839
 Summary: Efficient Triton kernels for LLM Training
 License: BSD 2-CLAUSE LICENSE
         Copyright 2024 LinkedIn Corporation 
@@ -45,6 +45,7 @@ Requires-Dist: datasets>=2.19.2; extra == "dev"
 Requires-Dist: seaborn; extra == "dev"
 Requires-Dist: mkdocs; extra == "dev"
 Requires-Dist: mkdocs-material; extra == "dev"
+Requires-Dist: torchvision>=0.20; extra == "dev"
 
 <a name="readme-top"></a>
 

{liger_kernel_nightly-0.5.10.dev20250611191801 → liger_kernel_nightly-0.5.10.dev20250611215839}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "liger_kernel_nightly"
-version = "0.5.10.dev20250611191801"
+version = "0.5.10.dev20250611215839"
 description = "Efficient Triton kernels for LLM Training"
 urls = { "Homepage" = "https://github.com/linkedin/Liger-Kernel" }
 readme = { file = "README.md", content-type = "text/markdown" }

{liger_kernel_nightly-0.5.10.dev20250611191801 → liger_kernel_nightly-0.5.10.dev20250611215839}/setup.py

@@ -43,6 +43,7 @@ def get_optional_dependencies():
             "seaborn",
             "mkdocs",
             "mkdocs-material",
+            "torchvision>=0.20",
         ]
     }
 

{liger_kernel_nightly-0.5.10.dev20250611191801 → liger_kernel_nightly-0.5.10.dev20250611215839}/src/liger_kernel_nightly.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: liger_kernel_nightly
-Version: 0.5.10.dev20250611191801
+Version: 0.5.10.dev20250611215839
 Summary: Efficient Triton kernels for LLM Training
 License: BSD 2-CLAUSE LICENSE
         Copyright 2024 LinkedIn Corporation 
@@ -45,6 +45,7 @@ Requires-Dist: datasets>=2.19.2; extra == "dev"
 Requires-Dist: seaborn; extra == "dev"
 Requires-Dist: mkdocs; extra == "dev"
 Requires-Dist: mkdocs-material; extra == "dev"
+Requires-Dist: torchvision>=0.20; extra == "dev"
 
 <a name="readme-top"></a>
 

{liger_kernel_nightly-0.5.10.dev20250611191801 → liger_kernel_nightly-0.5.10.dev20250611215839}/src/liger_kernel_nightly.egg-info/requires.txt

@@ -14,3 +14,4 @@ datasets>=2.19.2
 seaborn
 mkdocs
 mkdocs-material
+torchvision>=0.20

{liger_kernel_nightly-0.5.10.dev20250611191801 → liger_kernel_nightly-0.5.10.dev20250611215839}/test/convergence/bf16/test_mini_models_multimodal.py

@@ -20,6 +20,7 @@ from test.utils import FAKE_CONFIGS_PATH
 from test.utils import UNTOKENIZED_DATASET_PATH
 from test.utils import MiniModelConfig
 from test.utils import assert_verbose_allclose
+from test.utils import is_torchvision_available
 from test.utils import load_image_processing_config
 from test.utils import load_processor_config
 from test.utils import load_tokenizer_config
@@ -788,6 +789,7 @@ def run_mini_model_multimodal(
                     not QWEN2_VL_AVAILABLE,
                     reason="Qwen2-VL not available in this version of transformers",
                 ),
+                pytest.mark.skipif(not is_torchvision_available(), reason="Qwen2VLVideoProcessor requires torchvision"),
             ],
         ),
         pytest.param(
@@ -826,6 +828,7 @@ def run_mini_model_multimodal(
                     not QWEN2_5_VL_AVAILABLE,
                     reason="Qwen2.5-VL not available in this version of transformers",
                 ),
+                pytest.mark.skipif(not is_torchvision_available(), reason="Qwen2VLVideoProcessor requires torchvision"),
             ],
         ),
         pytest.param(

{liger_kernel_nightly-0.5.10.dev20250611191801 → liger_kernel_nightly-0.5.10.dev20250611215839}/test/convergence/fp32/test_mini_models_multimodal.py

@@ -20,6 +20,7 @@ from test.utils import FAKE_CONFIGS_PATH
 from test.utils import UNTOKENIZED_DATASET_PATH
 from test.utils import MiniModelConfig
 from test.utils import assert_verbose_allclose
+from test.utils import is_torchvision_available
 from test.utils import load_image_processing_config
 from test.utils import load_processor_config
 from test.utils import load_tokenizer_config
@@ -783,6 +784,7 @@ def run_mini_model_multimodal(
                     not QWEN2_VL_AVAILABLE,
                     reason="Qwen2-VL not available in this version of transformers",
                 ),
+                pytest.mark.skipif(not is_torchvision_available(), reason="Qwen2VLVideoProcessor requires torchvision"),
             ],
         ),
         pytest.param(
@@ -817,6 +819,7 @@ def run_mini_model_multimodal(
                     not QWEN2_5_VL_AVAILABLE,
                     reason="Qwen2.5-VL not available in this version of transformers",
                 ),
+                pytest.mark.skipif(not is_torchvision_available(), reason="Qwen2VLVideoProcessor requires torchvision"),
             ],
         ),
         pytest.param(

{liger_kernel_nightly-0.5.10.dev20250611191801 → liger_kernel_nightly-0.5.10.dev20250611215839}/test/utils.py

@@ -228,6 +228,13 @@ def supports_bfloat16():
         return False
 
 
+def is_torchvision_available():
+    if importlib.util.find_spec("torchvision") is not None:
+        return True
+    else:
+        return False
+
+
 def revert_liger_kernel_to_granite(model_config: MiniModelConfig):
     """
     Revert all Liger kernel patches applied to Granite.