run_glide.py 257 Bytes
Newer Older
anton-l's avatar
anton-l committed
1
import torch
anton-l's avatar
Style  
anton-l committed
2

3
from modeling_glide import GLIDE
anton-l's avatar
anton-l committed
4

anton-l's avatar
Style  
anton-l committed
5

anton-l's avatar
anton-l committed
6
7
8
9
generator = torch.Generator()
generator = generator.manual_seed(0)

# 1. Load models
10
pipeline = GLIDE.from_pretrained("fusing/glide-base")
anton-l's avatar
anton-l committed
11

anton-l's avatar
anton-l committed
12
img = pipeline("an oil painting of a corgi", generator)
anton-l's avatar
anton-l committed
13
14

print(img)