File size: 247 Bytes
8a6355e
992705a
8a6355e
 
992705a
8a6355e
1
2
3
4
5
6
import videogen_hub

model = videogen_hub.load('VideoCrafter2')
video = model.infer_one_video(prompt="A child excitedly swings on a rusty swing set, laughter filling the air.")

# Here video is a torch tensor of shape torch.Size([16, 3, 320, 512])