fix: type issue

docqai · Oct 23, 2023 · 03d174f · 03d174f
1 parent 757dade
commit 03d174f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/source/docq/support/metadata_extractors.py b/source/docq/support/metadata_extractors.py
@@ -238,7 +238,7 @@ def __init__(
         if device is not None:
             import torch
             tdevice = torch.device(device)
-            self._model = self._model.to(self, device=tdevice)
+            self._model = self._model.to(device=tdevice) # type: ignore
 
         self._tokenizer = tokenizer or word_tokenize