lint: add mypy ignore

This commit is contained in:
Limitex
2025-11-01 21:44:42 +09:00
parent 9234093723
commit 99947be37f
@@ -236,7 +236,7 @@ class DiffusersAutoencoderRepository(AutoencoderRepository):
device = "cuda" if torch.cuda.is_available() else "cpu"
if checkpoint_path.endswith("safetensors"):
checkpoint: dict[str, Any] = {}
with safe_open(checkpoint_path, framework="pt", device="cpu") as f: # type: ignore[no-untyped-call]
with safe_open(checkpoint_path, framework="pt", device="cpu") as f: # type: ignore[no-untyped-call, attr-defined]
for key in f:
checkpoint[key] = f.get_tensor(key)
else: