From 159d7b80f6a09144778b1f2cbf624a0f9932cb78 Mon Sep 17 00:00:00 2001
From: Vidminas Mikucionis <5411598+Vidminas@users.noreply.github.com>
Date: Sat, 2 Dec 2023 13:49:20 +0000
Subject: [PATCH] Nougat loader: download model to default cache dir

---
 chatdocs/document_loaders/nougat_loader.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/chatdocs/document_loaders/nougat_loader.py b/chatdocs/document_loaders/nougat_loader.py
index 201512e..46a5096 100644
--- 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():