Gyminfluencer / Stable Diffusion Script
Arabiansands's picture
Create Stable Diffusion Script
d5ad8f5 verified
Raw
History Blame Contribute Delete
407 Bytes
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")