Spaces:

Chaerin5
/

FoundHand

Runtime error

Chaerin5 commited on Dec 28, 2024

Commit

349b8de

1 Parent(s): 35c713b

enable zerogpu

Files changed (1) hide show

vqvae.py CHANGED Viewed

@@ -73,7 +73,6 @@ class Autoencoder(nn.Module):
         # Decode the image of shape `[batch_size, channels, height, width]`
         return self.decoder(z)
-    @spaces.GPU(duration=120)
     def forward(self, x):
         posterior = self.encode(x)
         z = posterior.sample()

         # Decode the image of shape `[batch_size, channels, height, width]`
         return self.decoder(z)
     def forward(self, x):
         posterior = self.encode(x)
         z = posterior.sample()