feat: ✨ PyTorch Recognition Model Multi-GPU support (#1164)

mindee · Mar 31, 2023 · 3b6649c · 3b6649c
1 parent 0f61524
commit 3b6649c
Show file tree

Hide file tree

Showing 2 changed files with 410 additions and 7 deletions.
diff --git a/doctr/models/factory/hub.py b/doctr/models/factory/hub.py
@@ -13,13 +13,7 @@
 from pathlib import Path
 from typing import Any
 
-from huggingface_hub import (  # type: ignore[attr-defined]
-    HfApi,
-    HfFolder,
-    Repository,
-    hf_hub_download,
-    snapshot_download,
-)
+from huggingface_hub import HfApi, HfFolder, Repository, hf_hub_download, snapshot_download
 
 from doctr import models
 from doctr.file_utils import is_tf_available, is_torch_available