mrleo1nid/gpu-rent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add Assistent reference books and remove training dataset output.

Browse Source 
Introduce books/ with civitai-krea2 and HF fictext builders, sha-diff seed to VM, and drop train.jsonl from the civitai scrape pipeline.

Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
Leonid Pershin
2026-08-23 22:56:07 +03:00
co-authored by Cursor
parent 84ed0bb47a
commit 07bb521b70
27 changed files with 836 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files
datasets/README.md
+10 -10
View File
@@ -1,13 +1,15 @@
# Civitai Krea2 datasets (local)
Scraped generation metadata for Assistent training + cheap FTS search. **No image files.**
Scraped generation metadata for Assistent **books** (FTS search). **No image files.** UI training (`train.jsonl`) removed.
Not part of `gpu-rent` Typer CLI and **not** `models.yaml` / `gpu-rent push`. Separate entrypoint:
Not part of `gpu-rent` Typer CLI and **not** `models.yaml` / `gpu-rent push`. Separate entrypoints:
```text
civitai-dataset.bat # Windows (or .\civitai-dataset.ps1 / ./civitai-dataset.sh)
# default = all → discover + scrape + split
# default = all → discover + scrape + split (+ books/civitai-krea2)
python -m gpu_rent.civitai_dataset all
books-dataset.bat # python -m gpu_rent.books download fictext …
```
Needs `CIVITAI_API_TOKEN` and preferably `CIVITAI_API_HOST=civitai.red` in `.env`.
@@ -18,7 +20,7 @@ Needs `CIVITAI_API_TOKEN` and preferably `CIVITAI_API_HOST=civitai.red` in `.env
| --- | --- | --- |
| `discover` | `--top-checkpoints` (20), `--top-loras` (30) | Popular Krea2 ckpt/LoRAs (+ versions from `models.yaml`) → `catalog/models.json` |
| `scrape` | `--target` (2000), `--per-version`, `--min-score` | Gallery metadata only; resume by id; NSFW→rating; skip minor tags |
| `split` | — | `by_kind/` / `by_rating/` / `train.jsonl` / `search.jsonl` |
| `split` | — | `by_kind/` / `by_rating/` / `search.jsonl` + `books/civitai-krea2/` |
| `all` | same as discover+scrape | discover → scrape → split (launcher default) |
Global: `--out` = app root (writes under `datasets/civitai`).
@@ -31,11 +33,9 @@ Global: `--out` = app root (writes under `datasets/civitai`).
| `civitai/catalog/images.jsonl` | Canonical rows (resume by id) |
| `civitai/by_kind/` | checkpoint vs lora |
| `civitai/by_rating/` | pg / pg13 / r / x |
| `civitai/train.jsonl` | Alpaca instruction/output for Assistent Обучение |
| `civitai/search.jsonl` | Compact FTS index → pushed to VM as `Assistent/civitai-examples.jsonl` on `up` / `seed-personas` |
| `civitai/search.jsonl` | Compact Civitai FTS source → `books/civitai-krea2` on split |
| `../books/civitai-krea2/` | Assistent book (`book.yaml`, `search.jsonl`, `meta.json`) |
Push of `search.jsonl` is skipped if the file is missing. Weights stay in [models.md](../docs/models.md) (Civitai seed / `Models/` / `capture`).
On `up` / `seed-personas` / `seed-books`: changed books push → `/mnt/swarm_data/Assistent/books/` (sha diff). Legacy `civitai-examples.jsonl` only if `books/` empty.
Git ignores the data; this README stays.
See also: [docs/cli.md](../docs/cli.md) (civitai-dataset block), [docs/extensions.md](../docs/extensions.md), [docs/local-folders.md](../docs/local-folders.md).
See [books/README.md](../books/README.md), [docs/cli.md](../docs/cli.md), [docs/local-folders.md](../docs/local-folders.md).