Spaces:
Runtime error
Runtime error
import torch | |
from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler | |
pipeline = StableDiffusionPipeline.from_pretrained( | |
"andite/anything-v4.0", torch_dtype=torch.float16, safety_checker=None | |
).to("cuda") | |
pipeline.scheduler = DPMSolverMultistepScheduler.from_config( | |
pipeline.scheduler.config, use_karras_sigmas=True | |
) | |
pipeline.load_lora_weights(".", weight_name="genshin.safetensors") | |
prompt = ("masterpiece, best quality, absurdres, 1girl, school uniform, kangel, smile, standing, contrapposto, " | |
"bedroom, leaning forward, <lora:kangelNeedyGirl_v10:1>, <lora:genshin-char-model:1>, genshin,") | |
negative_prompt = ("(low quality, worst quality:1.4), (bad anatomy), (inaccurate limb:1.2), " | |
"bad composition, inaccurate eyes, extra digit, fewer digits, (extra arms:1.2)") | |
images = pipeline(prompt=prompt, | |
negative_prompt=negative_prompt, | |
width=512, | |
height=512, | |
num_inference_steps=20, | |
num_images_per_prompt=1, | |
guidance_scale=7.0, | |
generator=torch.manual_seed(9625644) | |
).images | |
images[0].save("test.jpg") | |