bhuvanmdev
commited on
Update lgm/lgm.py
Browse files- lgm/lgm.py +1 -1
lgm/lgm.py
CHANGED
@@ -282,7 +282,7 @@ class LGM(ModelMixin, ConfigMixin):
|
|
282 |
x = self.unet(images,V) ###
|
283 |
x = self.conv(x)
|
284 |
|
285 |
-
x = x.reshape(B,
|
286 |
|
287 |
x = x.permute(0, 1, 3, 4, 2).reshape(B, -1, 14)
|
288 |
|
|
|
282 |
x = self.unet(images,V) ###
|
283 |
x = self.conv(x)
|
284 |
|
285 |
+
x = x.reshape(B, V, 14, self.splat_size, self.splat_size)
|
286 |
|
287 |
x = x.permute(0, 1, 3, 4, 2).reshape(B, -1, 14)
|
288 |
|