import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("klibaner/aquarium-fish-sdxl-lora")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]Aquarium Fish Classification LoRA (SDXL)
Эта LoRA обучена на датасете из 6 видов аквариумных обитателей. Модель предназначена для генерации фотореалистичных изображений рыб с помощью Stable Diffusion XL.
Триггерные слова (Trigger Words)
Для вызова конкретного класса используйте следующие токены в вашем промпте:
- Гуппи:
klm guppy fish - Оскар (Астронотус):
jpt oscar fish - Золотая рыбка:
vnt gold fish - Петушок (Betta):
zxw bete fish - Дискус:
qvp discuss fish - Рак:
brm cray
Пример использования (Prompt)
klm guppy fish, highly detailed macro photography, swimming in a beautifully decorated clear aquarium with neon lights, 4k resolution
- Downloads last month
- 32
Model tree for klibaner/aquarium-fish-sdxl-lora
Base model
stabilityai/stable-diffusion-xl-base-1.0