Fix AMP deprecation warnings

src/engines/coherence/engine.py

@@ -7,6 +7,7 @@ import tempfile
 import threading
 import time
 import urllib.request
+from contextlib import nullcontext
 from pathlib import Path
 from typing import Optional
 
@@ -194,6 +195,15 @@ def _prepare_runtime(device: str) -> None:
         _resnet_fallback.to(device)
 
 
+def _autocast_context(device: str):
+    if device != "cuda" or _torch is None:
+        return nullcontext()
+    try:
+        return _torch.amp.autocast("cuda")
+    except AttributeError:
+        return _torch.cuda.amp.autocast()
+
+
 def _release_runtime(device: str) -> None:
     global _device
     _device = "cpu"
@@ -479,7 +489,7 @@ class CoherenceEngine:
                     if face is not None:
                         face_gpu = face.unsqueeze(0).to(_device)
                         with _torch.no_grad():
-                            with _torch.cuda.amp.autocast(enabled=(_device == "cuda")):
+                            with _autocast_context(_device):
                                 emb = _resnet(face_gpu).detach().float().cpu().numpy()[0]
                         embeddings.append(emb)
                 except Exception:
@@ -515,7 +525,7 @@ class CoherenceEngine:
                 crop = Image.fromarray(frame[y1:y2, x1:x2]).convert("RGB")
                 tensor = _transform_fallback(crop).unsqueeze(0).to(_device)
                 with _torch.no_grad():
-                    with _torch.cuda.amp.autocast(enabled=(_device == "cuda")):
+                    with _autocast_context(_device):
                         emb = _resnet_fallback(tensor).detach().float().cpu().numpy()[0]
                 embeddings_fb.append(emb)
             except Exception:

src/engines/fingerprint/engine.py

@@ -10,6 +10,7 @@ import logging
 import os
 import threading
 import time
+from contextlib import nullcontext
 from typing import Any, Optional
 
 import numpy as np
@@ -175,6 +176,15 @@ def _prepare_runtime(device: str) -> None:
         _clip_model.to(device)
 
 
+def _autocast_context(device: str):
+    if device != "cuda":
+        return nullcontext()
+    try:
+        return torch.amp.autocast("cuda")
+    except AttributeError:
+        return torch.cuda.amp.autocast()
+
+
 def _release_runtime(device: str) -> None:
     if device != "cuda":
         return
@@ -285,7 +295,7 @@ class FingerprintEngine:
             inputs = {k: v.to(device) for k, v in inputs.items()}
 
             with torch.no_grad():
-                with torch.cuda.amp.autocast(enabled=(device == "cuda")):
+                with _autocast_context(device):
                     outputs = _clip_model(**inputs)
                     logits = outputs.logits_per_image[0].float()
                     image_embeds = outputs.image_embeds.detach().float().cpu().numpy()[0]
@@ -331,7 +341,7 @@ class FingerprintEngine:
             inputs = {k: v.to(device) for k, v in inputs.items()}
 
             with torch.no_grad():
-                with torch.cuda.amp.autocast(enabled=(device == "cuda")):
+                with _autocast_context(device):
                     logits = _clip_model(**inputs).logits_per_image.float()
 
             probs_batch = logits.softmax(dim=-1).cpu().numpy()