https://github.com/p1atdev/dartrs
https://github.com/p1atdev/dartrs
Last synced: 3 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/p1atdev/dartrs
- Owner: p1atdev
- License: apache-2.0
- Created: 2024-04-29T15:25:48.000Z (about 1 year ago)
- Default Branch: main
- Last Pushed: 2024-05-12T14:10:15.000Z (about 1 year ago)
- Last Synced: 2025-01-14T18:28:11.511Z (4 months ago)
- Language: Rust
- Size: 87.9 KB
- Stars: 2
- Watchers: 1
- Forks: 1
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# dartrs
> [!WARNING]
> wip
## Python
```bash
pip install -U dartrs
```
```py
from dartrs.dartrs import DartTokenizer
from dartrs.utils import get_generation_config
from dartrs.v2 import (
compose_prompt,
MixtralModel,
V2Model,
)
import time
import os
MODEL_NAME = "p1atdev/dart-v2-moe-sft"
model = MixtralModel.from_pretrained(MODEL_NAME, dtype="fp16")
tokenizer = DartTokenizer.from_pretrained(MODEL_NAME)
config = get_generation_config(
prompt=compose_prompt(
copyright="vocaloid",
character="hatsune miku",
rating="general", # sfw, general, sensitive, nsfw, questionable, explicit
aspect_ratio="tall", # ultra_wide, wide, square, tall, ultra_tall
length="medium", # very_short, short, medium, long, very_long
identity="none", # none, lax, strict
prompt="1girl, cat ears",
),
tokenizer=tokenizer,
seed=42,
)
start = time.time()
output = model.generate(config)
end = time.time()
print(output)
print(f"Time taken: {end - start:.2f}s")
# Output:
# cowboy shot, detached sleeves, expressionless, from side, hair ornament, halftone, hairclip, holding, limited palette, long hair, looking at viewer, miniskirt, necktie, pleated skirt, shirt, simple background, skirt, sleeveless, sleeveless shirt, straight-on, thighhighs, twintails, very long hair, zettai ryouiki
# Time taken: 0.22s
```