bhuvanmdev
commited on
Update pipeline.py
Browse files- 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 |
|