Bingsu/my-korean-stable-diffusion-v1-5와 같은 방법으로 만든 k-아무거나 3.0 모델.
생각보다 잘 안되고, 특히 캐릭터에 관한 정보는 다 잊어버린 걸로 보입니다.
Usage
pip install transformers accelerate>=0.14.0 diffusers>=0.7.2
import torch
from diffusers import StableDiffusionPipeline
repo = "Bingsu/my-k-anything-v3-0"
pipe = StableDiffusionPipeline.from_pretrained(
repo, torch_dtype=torch.float16,
)
pipe.to("cuda")
pipe.safety_checker = None
from typing import Optional
import torch
def gen_image(
prompt: str,
negative_prompt: Optional[str] = None,
seed: Optional[int] = None,
scale: float = 7.5,
steps: int = 30,
):
if seed is not None:
generator = torch.Generator("cuda").manual_seed(seed)
else:
generator = None
image = pipe(
prompt=prompt,
negative_prompt=negative_prompt,
generator=generator,
guidance_scale=scale,
num_inference_steps=steps,
).images[0]
return image
prompt = "파란색 포니테일 헤어, 브로치, 정장을 입은 성인 여성, 고퀄리티, 최고품질"
negative = "저화질, 저품질, 텍스트"
seed = 42467781
scale = 12.0
gen_image(prompt, negative, seed, scale)