Update float_lipsync.py

float_lipsync.py

@@ -28,23 +28,22 @@ import face_alignment
 import albumentations as A
 import albumentations.pytorch.transforms as A_pytorch
 from transformers import Wav2Vec2FeatureExtractor
-try:
-    import transformers.utils.import_utils as _tf_import
-    if hasattr(_tf_import, 'check_torch_load_is_safe'):
-        _tf_import.check_torch_load_is_safe = lambda: None
-        logger.info("[FLOAT] Patched transformers torch.load safety check for torch 2.5 compat")
-except Exception:
-    pass
 
 # Workaround: transformers now requires torch>=2.6 for torch.load safety (CVE-2025-32434)
 # but vLLM 0.7.3 pins torch 2.5.1. Disable the check since we only load trusted local checkpoints.
 try:
     import transformers.utils.import_utils as _tf_import
+    import transformers.modeling_utils as _tf_modeling
+    _noop = lambda: None
+    # Patch the source function
     if hasattr(_tf_import, 'check_torch_load_is_safe'):
-        _tf_import.check_torch_load_is_safe = lambda: None
-        logger.info("[FLOAT] Patched transformers torch.load safety check for torch 2.5 compat")
-except Exception:
-    pass
+        _tf_import.check_torch_load_is_safe = _noop
+    # Patch the local reference in modeling_utils (imported via 'from ... import')
+    if hasattr(_tf_modeling, 'check_torch_load_is_safe'):
+        _tf_modeling.check_torch_load_is_safe = _noop
+    logger.info("[FLOAT] Patched transformers torch.load safety check for torch 2.5 compat")
+except Exception as e:
+    logger.warning(f"[FLOAT] Failed to patch torch.load safety check: {e}")
 
 # Import FLOAT model from cloned repo
 from models.float.FLOAT import FLOAT
@@ -429,4 +428,4 @@ def get_lipsync() -> FloatLipsync:
     global _lipsync_instance
     if _lipsync_instance is None:
         _lipsync_instance = FloatLipsync()
-    return _lipsync_instance
+    return _lipsync_instance