Mitsua/vroid-image-dataset-lite
Viewer โข Updated โข 3k โข 55 โข 11
How to use Mitsua/vroid-diffusion-test with Diffusers:
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("Mitsua/vroid-diffusion-test", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]This is a latent text-to-image diffusion model to demonstrate how U-Net training affects the generated images.
vroid_diffusion_test.safetensorsvroid_diffusion_test_invert_red_blue.safetensorsred and blue in the caption is swapped.pink and skyblue in the caption is swapped.vroid_diffusion_test_monochrome.safetensorsText-to-Image generation for research and educational purposes.
Any deployed use case of the model.
We use full version of VRoid Image Dataset Lite with some modifications.