2022-06-07 04:19:53 -06:00
|
|
|
import torch
|
|
|
|
|
from .modeling_glide import GLIDE
|
|
|
|
|
from diffusers import UNetGLIDEModel, GaussianDDPMScheduler
|
|
|
|
|
|
|
|
|
|
generator = torch.Generator()
|
|
|
|
|
generator = generator.manual_seed(0)
|
|
|
|
|
|
|
|
|
|
# 1. Load models
|
|
|
|
|
scheduler = GaussianDDPMScheduler.from_config("fusing/glide-base")
|
|
|
|
|
model = UNetGLIDEModel.from_pretrained("fusing/glide-base")
|
|
|
|
|
|
|
|
|
|
pipeline = GLIDE(model, scheduler)
|
|
|
|
|
|
|
|
|
|
img = pipeline(generator)
|
|
|
|
|
|
|
|
|
|
print(img)
|
