run_glide.py 396 Bytes
Newer Older
anton-l's avatar
anton-l committed
1
2
3
4
5
6
7
8
import torch
from .modeling_glide import GLIDE
from diffusers import UNetGLIDEModel, GaussianDDPMScheduler

generator = torch.Generator()
generator = generator.manual_seed(0)

# 1. Load models
anton-l's avatar
anton-l committed
9

anton-l's avatar
anton-l committed
10
11
12
13
14
15
16
17
scheduler = GaussianDDPMScheduler.from_config("fusing/glide-base")
model = UNetGLIDEModel.from_pretrained("fusing/glide-base")

pipeline = GLIDE(model, scheduler)

img = pipeline(generator)

print(img)