, weights_only=True

hyvideo/modules/fp8_optimization.py

@@ -85,7 +85,7 @@ def convert_fp8_linear(module, dit_weight_path, original_dtype, params_to_keep={
     # loading fp8 mapping file
     fp8_map_path = dit_weight_path.replace(".pt", "_map.pt")
     if os.path.exists(fp8_map_path):
-        fp8_map = torch.load(fp8_map_path, map_location=lambda storage, loc: storage)
+        fp8_map = torch.load(fp8_map_path, weights_only=True, map_location=lambda storage, loc: storage)
     else:
         raise ValueError(f"Invalid fp8_map path: {fp8_map_path}.")