jalr/tabbyAPI-ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Downloader: Fix handling of include pattern

Browse source 
If an include or exclude pattern is provided, include should include
all files by default.

Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
kingbri 2024-04-30 01:13:06 -04:00
parent 21a01741c9
commit 50e0b71690
2 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
common/downloader.py
View file

@ -121,7 +121,7 @@ async def hf_repo_download(
repo_type = "lora"
if include or exclude:
include_patterns = unwrap(include, [])
include_patterns = unwrap(include, ["*"])
exclude_patterns = unwrap(exclude, [])
file_list = [

6
endpoints/OAI/types/download.py
View file

@ -2,6 +2,10 @@ from pydantic import BaseModel, Field
from typing import List, Optional
def _generate_include_list():
return ["*"]
class DownloadRequest(BaseModel):
"""Parameters for a HuggingFace repo download."""
@ -10,7 +14,7 @@ class DownloadRequest(BaseModel):
folder_name: Optional[str] = None
revision: Optional[str] = None
token: Optional[str] = None
include: List[str] = Field(default_factory=list)
include: List[str] = Field(default_factory=_generate_include_list)
exclude: List[str] = Field(default_factory=list)
chunk_limit: Optional[int] = None