require torch 2, allow numpy>2, remove unnecessary warnings

Signed-off-by: Brian Dellabetta <[email protected]>

Author: brian-dellabetta

setup.py

@@ -112,10 +112,10 @@ def localversion_func(version: ScmVersion) -> str:
     install_requires=[
         "loguru",
         "pyyaml>=5.0.0",
-        "numpy>=1.17.0,<2.0",
+        "numpy>=1.17.0",
         "requests>=2.0.0",
         "tqdm>=4.0.0",
-        "torch>=1.7.0",
+        "torch>=2.7.0",
         "transformers>4.0,<5.0",
         "datasets",
         "accelerate>=0.20.3,!=1.1.0",

src/llmcompressor/pytorch/__init__.py

@@ -1,45 +0,0 @@
-"""
-Functionality for working with and sparsifying Models in the PyTorch framework
-"""
-
-import os
-import warnings
-
-from packaging import version
-
-try:
-    import torch
-
-    _PARSED_TORCH_VERSION = version.parse(torch.__version__)
-
-    if _PARSED_TORCH_VERSION.major >= 2:
-        torch_compile_func = torch.compile
-
-        def raise_torch_compile_warning(*args, **kwargs):
-            warnings.warn(
-                "torch.compile is not supported by llmcompressor for torch 2.0.x"
-            )
-            return torch_compile_func(*args, **kwargs)
-
-        torch.compile = raise_torch_compile_warning
-
-    _BYPASS = bool(int(os.environ.get("NM_BYPASS_TORCH_VERSION", "0")))
-    if _PARSED_TORCH_VERSION.major == 1 and _PARSED_TORCH_VERSION.minor in [10, 11]:
-        if not _BYPASS:
-            raise RuntimeError(
-                "llmcompressor does not support torch==1.10.* or 1.11.*. "
-                f"Found torch version {torch.__version__}.\n\n"
-                "To bypass this error, set environment variable "
-                "`NM_BYPASS_TORCH_VERSION` to '1'.\n\n"
-                "Bypassing may result in errors or "
-                "incorrect behavior, so set at your own risk."
-            )
-        else:
-            warnings.warn(
-                "llmcompressor quantized onnx export does not work "
-                "with torch==1.10.* or 1.11.*"
-            )
-except ImportError:
-    pass
-
-# flake8: noqa