29 - Added compatibility with single file downloads

Signed-off-by: julianbollig <[email protected]>
TNG · Nov 29, 2024 · 96b98d0 · 96b98d0
1 parent 9df3fe2
commit 96b98d0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/service/model_downloader.py b/service/model_downloader.py
@@ -334,6 +334,7 @@ def init_download(self, file: HFDonloadItem):
 
     def is_access_granted(self, repo_id: str, model_type, backend : str):
 
+        repo_id = utils.trim_repo(repo_id)
         headers={}
         if (self.hf_token is not None):
             headers["Authorization"] = f"Bearer {self.hf_token}"