Nougat loader: download model to default cache dir

Vidminas · Dec 2, 2023 · 159d7b8 · 159d7b8
1 parent 4832b35
commit 159d7b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/chatdocs/document_loaders/nougat_loader.py b/chatdocs/document_loaders/nougat_loader.py
@@ -34,7 +34,7 @@ def __init__(
                 "`nougat` package not found, please install it with "
                 "`pip install nougat-ocr`"
             )
-        checkpoint = get_checkpoint("nougat", download=True)
+        checkpoint = get_checkpoint(download=True)
         self.model = NougatModel.from_pretrained(checkpoint)
 
         if torch.cuda.is_available():