File size: 839 Bytes
798f776
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
from diffusers import StableDiffusionXLPipeline
import torch
import rembg

# text-to-image with SDXL for text-to-image-to-3d
pipeline = StableDiffusionXLPipeline.from_single_file(
    "https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/blob/main/sd_xl_base_1.0.safetensors",
    torch_dtype=torch.float16,
    variant="fp16",
    use_safetensors=True
).to("cuda")
pipeline.enable_model_cpu_offload()

num_images_per_prompt = 1
res = 1024
text = input("Prompt > ")
bkgd_color = "white"
prompt = f"a ((full-body:2)) shot of a ((single:2)) {text}, isolated on {bkgd_color} background, 4k, highly detailed"
images = pipeline(prompt=prompt, num_images_per_prompt=num_images_per_prompt, height=res, width=res).images
image = images[0]
image.show()
image.save("output.png")
image = rembg.remove(image)
image.save("output-rembg.png")