from diffusers import DiffusionPipeline | |
pipeline = DiffusionPipeline.from_pretrained("miliyas/practise") | |
from diffusers import AutoPipelineForText2Image | |
import torch | |
pipeline = AutoPipelineForText2Image.from_pretrained( | |
"runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16 | |
).to("cuda") | |
image = pipeline( | |
prompt="Astronaut in a jungle, cold color palette, muted colors, detailed, 8k", | |
negative_prompt="ugly, deformed, disfigured, poor details, bad anatomy", | |
).images[0] | |
image |