Spaces:
Running
Running
dev(narugo): quick webp
Browse files
pools.py
CHANGED
@@ -2,20 +2,6 @@ from typing import Type
|
|
2 |
|
3 |
from cheesechaser.datapool import IncrementIDDataPool
|
4 |
|
5 |
-
_AIBOORU_REPO_ID = 'deepghs/aibooru-webp-4Mpixel'
|
6 |
-
|
7 |
-
|
8 |
-
class AIBooruWebpDataPool(IncrementIDDataPool):
|
9 |
-
def __init__(self, revision: str = 'main'):
|
10 |
-
IncrementIDDataPool.__init__(
|
11 |
-
self,
|
12 |
-
data_repo_id=_AIBOORU_REPO_ID,
|
13 |
-
data_revision=revision,
|
14 |
-
idx_repo_id=_AIBOORU_REPO_ID,
|
15 |
-
idx_revision=revision,
|
16 |
-
base_level=3,
|
17 |
-
)
|
18 |
-
|
19 |
|
20 |
def quick_webp_pool(site_name: str, level: int = 3) -> Type[IncrementIDDataPool]:
|
21 |
repo_id = f'deepghs/{site_name}-webp-4Mpixel'
|
|
|
2 |
|
3 |
from cheesechaser.datapool import IncrementIDDataPool
|
4 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
5 |
|
6 |
def quick_webp_pool(site_name: str, level: int = 3) -> Type[IncrementIDDataPool]:
|
7 |
repo_id = f'deepghs/{site_name}-webp-4Mpixel'
|