"vscode:/vscode.git/clone" did not exist on "75aee39eac7a13316b1723d94d78c6d84b1cebd4"
Diffusers.py 517 Bytes
Newer Older
lijian6's avatar
lijian6 committed
1
from diffusers import AutoPipelineForText2Image
lijian6's avatar
lijian6 committed
2
import torch
lijian6's avatar
lijian6 committed
3
import time
lijian6's avatar
lijian6 committed
4

lijian6's avatar
lijian6 committed
5
6
7
pipeline = AutoPipelineForText2Image.from_pretrained(
	"stable-diffusion-v1-5", torch_dtype=torch.float16, variant="fp16"
).to("cuda")
lijian6's avatar
lijian6 committed
8

lijian6's avatar
lijian6 committed
9
10
11
12
13
14
generator = torch.Generator("cuda").manual_seed(31)
for i in range(2):
    start = time.time()
    image = pipeline("Astronaut in a jungle, cold color palette, muted colors, detailed, 8k", generator=generator).images[0]
    image.save(f"astronaut_rides_horse{i}.png")
    print(f"time:{time.time()-start}")