bhuvanmdev commited on
Commit
5c627cb
·
verified ·
1 Parent(s): 8d62889

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -1613,7 +1613,7 @@ class LGMFullPipeline(DiffusionPipeline):
1613
  images, self.imagenet_default_mean, self.imagenet_default_std
1614
  )
1615
 
1616
- rays_embeddings = self.lgm.prepare_default_rays("cuda", elevation=0)
1617
  images = torch.cat([images, rays_embeddings], dim=1).unsqueeze(0)
1618
  images = images.half().cuda()
1619
 
 
1613
  images, self.imagenet_default_mean, self.imagenet_default_std
1614
  )
1615
 
1616
+ rays_embeddings = self.lgm.prepare_default_rays("cuda", elevation=0,views=num_frames)
1617
  images = torch.cat([images, rays_embeddings], dim=1).unsqueeze(0)
1618
  images = images.half().cuda()
1619