autogluon.multimodal 1.2.1b20250407__py3-none-any.whl → 1.2.1b20250409__py3-none-any.whl

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/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/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.dist-info → autogluon.multimodal-1.2.1b20250409.dist-info}/METADATA

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

{autogluon.multimodal-1.2.1b20250407.dist-info → autogluon.multimodal-1.2.1b20250409.dist-info}/RECORD

@@ -1,8 +1,8 @@
-autogluon.multimodal-1.2.1b20250407-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.multimodal-1.2.1b20250409-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
 autogluon/multimodal/__init__.py,sha256=6WuDt3sMP1QLJs9se-20mmHCsFR-q2ZttPc8s0k1QWA,204
 autogluon/multimodal/constants.py,sha256=eLF3t-447nw3aNrYPh4Y8pycdECGu__wv6TC-amfXw8,9509
 autogluon/multimodal/predictor.py,sha256=4lou5yGysY1O86A-PX8AgsJvjB5bq2eHJ9zB1DyZFew,42847
-autogluon/multimodal/version.py,sha256=Hl84ugmOnqfU18pLVhKVOks3lzI2_-__H8uz6UWTQvY,91
+autogluon/multimodal/version.py,sha256=A9BqrXhLrOJhH5qPbhMof-jLGxRS5nRGGDYPGWeziJE,91
 autogluon/multimodal/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/multimodal/cli/prepare_detection_dataset.py,sha256=9NCYmCUMPRWbxxbN_C7YQjMYlrAm8gbwJ3Qd-79JWH4,5218
 autogluon/multimodal/cli/voc2coco.py,sha256=LXVu9RIfOZs_1URrzO-_3Nrvb9uGEgPxCY4-B6m1coc,9605
@@ -65,7 +65,7 @@ autogluon/multimodal/data/process_mmlab/process_mmdet.py,sha256=wWcoeLoostH4y9oi
 autogluon/multimodal/data/process_mmlab/process_mmlab_base.py,sha256=7RorcwDoMRfLgtFEmH26uEg0csKvxyD21gdPh-9N4YM,6540
 autogluon/multimodal/data/process_mmlab/process_mmocr.py,sha256=dwhPDq1A84eqwZHnQMZFyg9GcnFTEkMCLBhJfhUeqQk,2913
 autogluon/multimodal/learners/__init__.py,sha256=BnOY7nwfXJ6rjPawF1mebSxB5Jx-OdiGwc_P53kNXOc,294
-autogluon/multimodal/learners/base.py,sha256=CkEq1RispBsW99vpDLuBIVU3_KwIc3yuHiEneF4BKww,99600
+autogluon/multimodal/learners/base.py,sha256=YzJnVr_FamfJa9sSEQq4n7ekSz0QFjeWuZRMVAMiUSM,99640
 autogluon/multimodal/learners/ensemble.py,sha256=yJOyGdgeqKeriSCdiv809pn8AEdx6uGIKG5O5RGp_wY,30525
 autogluon/multimodal/learners/few_shot_svm.py,sha256=XWD7uufpyemGSM9z8rIXksbHvU3YRlgQ0Vq_Wm0Sxe0,23919
 autogluon/multimodal/learners/matching.py,sha256=l0gXRAECBlO7G9_pYSaytl3RDERngMNLm7pzxUcPYwg,88428
@@ -126,7 +126,7 @@ autogluon/multimodal/optim/metrics/semantic_seg_metrics.py,sha256=tIbSk3iyBRRx7H
 autogluon/multimodal/optim/metrics/utils.py,sha256=VU7MJbJplgl89aYqyKrs2WbX-BkU5_0vYGWpPnTjJDQ,12738
 autogluon/multimodal/utils/__init__.py,sha256=WBfesTQWIpeH1EDxPWgPqr_mUk_X9g-WW-rgwmtqvOE,2543
 autogluon/multimodal/utils/cache.py,sha256=USg-uBtrhrxKlf0OMz-xbM67A8olfSBRAWU4VYuT4Vg,8180
-autogluon/multimodal/utils/checkpoint.py,sha256=aF65wkLzMS6ng4YE_1LZuCbbjaNudn0Cfi0insm8I6Q,9745
+autogluon/multimodal/utils/checkpoint.py,sha256=OvD6Bkj6XCwkdDkOKsrPcEwUNGEteivaRBrSr4mH3EI,9805
 autogluon/multimodal/utils/colormap.py,sha256=DOSPCgeQXk87B2ae3iM7T0RGjrIVozvwp7RHEXzyb-4,3882
 autogluon/multimodal/utils/config.py,sha256=b-vNiCzmVezygBSkGTtGRyFmVXdznxQIm6P6O_tKi-w,32518
 autogluon/multimodal/utils/device.py,sha256=60g5-tKCVnn_LAvxG_dnS5cZEzgzLTwU-A4gd-uvIfg,3523
@@ -153,11 +153,11 @@ autogluon/multimodal/utils/registry.py,sha256=vqvONWweZP44aBo1jCvlqLdL0Agn2kyTK8
 autogluon/multimodal/utils/save.py,sha256=aXZa_iue34dAEfTz7nCaRowktG1emEi5uVXe_tDmHBA,4408
 autogluon/multimodal/utils/strategy.py,sha256=tT9PWh_ZLwNdGFgPsXgZsgKRhpnfBQDjh1mB1_y8G18,833
 autogluon/multimodal/utils/visualizer.py,sha256=qAc4_36r3MBDPq1R1jBeb_gcfzIrsylL1S31sRf3wuw,22562
-autogluon.multimodal-1.2.1b20250407.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.multimodal-1.2.1b20250407.dist-info/METADATA,sha256=JuhvoMA5lfVc8bCOfoCcWO34Z21gPC0jJGKVYaCBPzs,13221
-autogluon.multimodal-1.2.1b20250407.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.multimodal-1.2.1b20250407.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.multimodal-1.2.1b20250407.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.multimodal-1.2.1b20250407.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.multimodal-1.2.1b20250407.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.multimodal-1.2.1b20250407.dist-info/RECORD,,
+autogluon.multimodal-1.2.1b20250409.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.multimodal-1.2.1b20250409.dist-info/METADATA,sha256=b9pjfmxED3_ZjOVsKMMLMXIzbE7CYoy3LD2gUTMWztw,13221
+autogluon.multimodal-1.2.1b20250409.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.multimodal-1.2.1b20250409.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.multimodal-1.2.1b20250409.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.multimodal-1.2.1b20250409.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.multimodal-1.2.1b20250409.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.multimodal-1.2.1b20250409.dist-info/RECORD,,