Spaces:
Running
on
Zero
Running
on
Zero
enable zerogpu
Browse files
vqvae.py
CHANGED
|
@@ -73,7 +73,6 @@ class Autoencoder(nn.Module):
|
|
| 73 |
# Decode the image of shape `[batch_size, channels, height, width]`
|
| 74 |
return self.decoder(z)
|
| 75 |
|
| 76 |
-
@spaces.GPU(duration=120)
|
| 77 |
def forward(self, x):
|
| 78 |
posterior = self.encode(x)
|
| 79 |
z = posterior.sample()
|
|
|
|
| 73 |
# Decode the image of shape `[batch_size, channels, height, width]`
|
| 74 |
return self.decoder(z)
|
| 75 |
|
|
|
|
| 76 |
def forward(self, x):
|
| 77 |
posterior = self.encode(x)
|
| 78 |
z = posterior.sample()
|