File size: 407 Bytes
d5ad8f5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
from diffusers import StableDiffusionPipeline
import torch

model_id = "stabilityai/stable-diffusion-xl-base-1.0"

pipe = StableDiffusionPipeline.from_pretrained(
    model_id,
    torch_dtype=torch.float16
).to("cuda")

prompt = "beautiful indian AI influencer girl, luxury fashion, cinematic lighting, ultra realistic, instagram influencer"

image = pipe(prompt).images[0]

image.save("ai_influencer.png")