run_glide.py 227 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
12
13
14

img = pipeline(generator)

print(img)