autogluon.multimodal 1.2.1b20250407__tar.gz → 1.2.1b20250409__tar.gz

{autogluon.multimodal-1.2.1b20250407 → autogluon.multimodal-1.2.1b20250409}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.multimodal
-Version: 1.2.1b20250407
+Version: 1.2.1b20250409
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community

{autogluon.multimodal-1.2.1b20250407 → autogluon.multimodal-1.2.1b20250409}/setup.py

@@ -37,7 +37,7 @@ install_requires = [
     "seqeval>=1.2.2,<1.3.0",
     "evaluate>=0.4.0,<0.5.0",
     "timm>=0.9.5,<1.0.7",
-    "torchvision>=0.16.0,<0.21.0",
+    "torchvision>=0.16.0,<0.22.0",
     "scikit-image>=0.19.1,<0.25.0",
     "text-unidecode>=1.3,<1.4",
     "torchmetrics>=1.2.0,<1.6.2",

{autogluon.multimodal-1.2.1b20250407 → autogluon.multimodal-1.2.1b20250409}/src/autogluon/multimodal/learners/base.py

@@ -2145,9 +2145,9 @@ class BaseLearner(ExportMixin, DistillationMixin, RealtimeMixin):
 
                 convert_zero_checkpoint_to_fp32_state_dict(path + "-dir", path)
                 shutil.rmtree(path + "-dir")
-                state_dict = torch.load(path, map_location=torch.device("cpu"))["state_dict"]  # nosec B614
+                state_dict = torch.load(path, map_location=torch.device("cpu"), weights_only=False)["state_dict"]  # nosec B614
             else:
-                state_dict = torch.load(path, map_location=torch.device("cpu"))["state_dict"]  # nosec B614
+                state_dict = torch.load(path, map_location=torch.device("cpu"), weights_only=False)["state_dict"]  # nosec B614
         state_dict = {k.partition(prefix)[2]: v for k, v in state_dict.items() if k.startswith(prefix)}
 
         # Some buffers like `position_ids` are registered as persistent=False since transformers 4.31.0

{autogluon.multimodal-1.2.1b20250407 → autogluon.multimodal-1.2.1b20250409}/src/autogluon/multimodal/utils/checkpoint.py

@@ -54,9 +54,9 @@ def average_checkpoints(
 
                 convert_zero_checkpoint_to_fp32_state_dict(per_path + "-dir", per_path)
                 shutil.rmtree(per_path + "-dir")
-                state_dict = torch.load(per_path, map_location=torch.device("cpu"))["state_dict"]  # nosec B614
+                state_dict = torch.load(per_path, map_location=torch.device("cpu"), weights_only=False)["state_dict"]  # nosec B614
             else:
-                state_dict = torch.load(per_path, map_location=torch.device("cpu"))["state_dict"]  # nosec B614
+                state_dict = torch.load(per_path, map_location=torch.device("cpu"), weights_only=False)["state_dict"]  # nosec B614
             for k, v in state_dict.items():
                 if k not in avg_state_dict:
                     avg_state_dict[k] = v.clone().to(dtype=torch.float64)
@@ -74,7 +74,7 @@ def average_checkpoints(
         for k in avg_state_dict:
             avg_state_dict[k].clamp_(float32_info.min, float32_info.max).to(dtype=torch.float32)
     else:
-        avg_state_dict = torch.load(checkpoint_paths[0], map_location=torch.device("cpu"))["state_dict"]  # nosec B614
+        avg_state_dict = torch.load(checkpoint_paths[0], map_location=torch.device("cpu"), weights_only=False)["state_dict"]  # nosec B614
 
     return avg_state_dict
 

{autogluon.multimodal-1.2.1b20250407 → autogluon.multimodal-1.2.1b20250409}/src/autogluon/multimodal/version.py

@@ -1,4 +1,4 @@
 """This is the autogluon version file."""
 
-__version__ = "1.2.1b20250407"
+__version__ = "1.2.1b20250409"
 __lite__ = False

{autogluon.multimodal-1.2.1b20250407 → autogluon.multimodal-1.2.1b20250409}/src/autogluon.multimodal.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.multimodal
-Version: 1.2.1b20250407
+Version: 1.2.1b20250409
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community

{autogluon.multimodal-1.2.1b20250407 → autogluon.multimodal-1.2.1b20250409}/src/autogluon.multimodal.egg-info/requires.txt

@@ -5,23 +5,23 @@ scikit-learn<1.5.3,>=1.4.0
 Pillow<12,>=10.0.1
 tqdm<5,>=4.38
 boto3<2,>=1.10
-torch<2.6,>=2.2
-lightning<2.6,>=2.2
+torch<2.7,>=2.2
+lightning<2.7,>=2.2
 transformers[sentencepiece]<4.50,>=4.38.0
-accelerate<1.0,>=0.34.0
+accelerate<2.0,>=0.34.0
 requests<3,>=2.30
 jsonschema<4.22,>=4.18
 seqeval<1.3.0,>=1.2.2
 evaluate<0.5.0,>=0.4.0
 timm<1.0.7,>=0.9.5
-torchvision<0.21.0,>=0.16.0
+torchvision<0.22.0,>=0.16.0
 scikit-image<0.25.0,>=0.19.1
 text-unidecode<1.4,>=1.3
 torchmetrics<1.6.2,>=1.2.0
 omegaconf<2.3.0,>=2.1.1
-autogluon.core[raytune]==1.2.1b20250407
-autogluon.features==1.2.1b20250407
-autogluon.common==1.2.1b20250407
+autogluon.core[raytune]==1.2.1b20250409
+autogluon.features==1.2.1b20250409
+autogluon.common==1.2.1b20250409
 pytorch-metric-learning<2.4,>=1.3.0
 nlpaug<1.2.0,>=1.1.10
 nltk<3.9,>=3.4.5