File size: 419 Bytes
e72f6f2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
# this is the huggingface inference script for the animation pipeline
# it takes in a text prompt and generates an animation based on the text

import os
import sys
import json
import torch
import argparse
import numpy as np
from PIL import Image
from tqdm import tqdm

import torch.nn.functional as F
from torch.utils.data import DataLoader
from torchvision import transforms
from torchvision.utils import save_image