narugo's picture
dev(narugo): quick webp
b4053ed
raw
history blame contribute delete
614 Bytes
from typing import Type
from cheesechaser.datapool import IncrementIDDataPool
def quick_webp_pool(site_name: str, level: int = 3) -> Type[IncrementIDDataPool]:
repo_id = f'deepghs/{site_name}-webp-4Mpixel'
class _QuickWebpDataPool(IncrementIDDataPool):
def __init__(self, revision: str = 'main'):
IncrementIDDataPool.__init__(
self,
data_repo_id=repo_id,
data_revision=revision,
idx_repo_id=repo_id,
idx_revision=revision,
base_level=level,
)
return _QuickWebpDataPool