github-starred/KohakuHub
2
0
Fork 0
mirror of https://github.com/KohakuBlueleaf/KohakuHub.git synced 2026-04-30 17:37:51 -05:00
Code Issues 5 Packages Projects Releases Wiki Activity

add lfs multipart setting into config

Browse Source
This commit is contained in:
Kohaku-Blueleaf
2025-10-21 21:56:52 +08:00
parent 8ac885f516
commit a4743c6133
3 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docker-compose.example.yml
View File

@@ -59,6 +59,8 @@ services:
## ===== Application Configuration ===== ## ===== Application Configuration =====
- KOHAKU_HUB_SITE_NAME=KohakuHub - KOHAKU_HUB_SITE_NAME=KohakuHub
- KOHAKU_HUB_LFS_THRESHOLD_BYTES=1000000 - KOHAKU_HUB_LFS_THRESHOLD_BYTES=1000000
- KOHAKU_HUB_LFS_MULTIPART_THRESHOLD_BYTES=100_000_000 # 100MB - use multipart for files larger than this
- KOHAKU_HUB_LFS_MULTIPART_CHUNK_SIZE_BYTES=50_000_000 # 50MB - size of each part (min 5MB except last)
- KOHAKU_HUB_LFS_KEEP_VERSIONS=5 - KOHAKU_HUB_LFS_KEEP_VERSIONS=5
- KOHAKU_HUB_LFS_AUTO_GC=true - KOHAKU_HUB_LFS_AUTO_GC=true

2
scripts/generate_docker_compose.py
View File

@@ -297,6 +297,8 @@ def generate_hub_api_service(config: dict) -> str:
## ===== Application Configuration ===== ## ===== Application Configuration =====
- KOHAKU_HUB_SITE_NAME=KohakuHub - KOHAKU_HUB_SITE_NAME=KohakuHub
- KOHAKU_HUB_LFS_THRESHOLD_BYTES=1000000 - KOHAKU_HUB_LFS_THRESHOLD_BYTES=1000000
- KOHAKU_HUB_LFS_MULTIPART_THRESHOLD_BYTES=100_000_000 # 100MB - use multipart for files larger than this
- KOHAKU_HUB_LFS_MULTIPART_CHUNK_SIZE_BYTES=50_000_000 # 50MB - size of each part (min 5MB except last)
- KOHAKU_HUB_LFS_KEEP_VERSIONS=5 - KOHAKU_HUB_LFS_KEEP_VERSIONS=5
- KOHAKU_HUB_LFS_AUTO_GC=true - KOHAKU_HUB_LFS_AUTO_GC=true
- KOHAKU_HUB_AUTO_MIGRATE=true # Auto-confirm database migrations (required for Docker) - KOHAKU_HUB_AUTO_MIGRATE=true # Auto-confirm database migrations (required for Docker)

6
src/kohakuhub/config.py
View File

@@ -331,6 +331,12 @@ def load_config(path: str = None) -> Config:
lfs_threshold_bytes=int( lfs_threshold_bytes=int(
os.environ.get("KOHAKU_HUB_LFS_THRESHOLD_BYTES", "5242880") os.environ.get("KOHAKU_HUB_LFS_THRESHOLD_BYTES", "5242880")
), ),
lfs_multipart_threshold_bytes=int(
os.environ.get("KOHAKU_HUB_LFS_MULTIPART_THRESHOLD_BYTES", "104857600")
),
lfs_multipart_chunk_size_bytes=int(
os.environ.get("KOHAKU_HUB_LFS_MULTIPART_CHUNK_SIZE_BYTES", "52428800")
),
lfs_keep_versions=int(os.environ.get("KOHAKU_HUB_LFS_KEEP_VERSIONS", "5")), lfs_keep_versions=int(os.environ.get("KOHAKU_HUB_LFS_KEEP_VERSIONS", "5")),
lfs_auto_gc=os.environ.get("KOHAKU_HUB_LFS_AUTO_GC", "false").lower() lfs_auto_gc=os.environ.get("KOHAKU_HUB_LFS_AUTO_GC", "false").lower()
== "true", == "true",