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") | |