import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("Tramac/style-portrait-v1-5", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]Style Portrait
Style Portrait is a SD-v1.5 based text-to-image generation model. It can generate high-quality 512*832 images of Asia portrait.
This repository contains checkpoints for diffusers and SD-WebUI.
Checkpoints
- style-portait-v1-5-fp16.safetensors: Checkpoint for Stable-Diffusion-WebUI.
- Others: Checkpoint for Diffusers.
Diffusers Usage
import torch
from diffusers import DiffusionPipeline
model_id = "Tramac/style-portrait-v1-5"
pipeline = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
pipeline = pipeline.to("cuda")
prompt = "1girl, long hair, close-up, elegant, upper body, natural skin texture, highly detailed"
negative_prompt = "cloned face, ugly, out of frame, mutated, extra arms, cross-eye, blurry, bad art, bad anatomy, badhands, watermark, text, long neck, worst quality, low quality"
image = pipeline(
prompt=prompt,
negative_prompt=negative_prompt,
height=832,
width=512,
).images[0]
image.save("demo.png")
WebUI Usage
- Download the checkpoint (style-portait-v1-5-fp16.safetensors) to /Stable-Diffusion-WebUI/models/Stable-diffusion.
- Start up WebUI and load checkpoint.
- Downloads last month
- 4