import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("minusthousand/lv-sb", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]Stable Diffision v1.5 fine tuned model with Latvian concepts.
Colab link with NLLB-200 translation: sb-lv-colab
Concept list (tokens):
- House of the Black Heads (melgalvjunams)
- National Library of Latvia (nacbib)
- The Vanšu Bridge (vansutilts)
- The Freedom Monument (brivpiem)
- University of Latvia (latvijasuniversitate, lvuni)
- Ginger snap (piparkuka)
- Riga Black Balsam (rigabalsam)
- Latvian traditional symbols (latviesurakstuzimes)
- Latvian Song and Dance Festival (dziesmusvetki)
- Rainis (rainis)
Sample images from the model:
prompt: a photo of rainis writing a book
prompt: a photo of rigabalsam and piparkuka
prompt: a photo of brivpiem statue
prompt: a photo of latvijasuniversitate building
- Downloads last month
- 20