.

app.py

@@ -126,7 +126,7 @@ pipe.to(device=DEVICE)
 #pipe.unet = torch.compile(pipe.unet)
 #pipe.vae = torch.compile(pipe.vae)
 
-
+@spaces.GPU()
 def generate_gpu(in_im_embs):
     print('start gen')
     in_im_embs = in_im_embs.to('cuda').unsqueeze(0).unsqueeze(0)
@@ -139,7 +139,7 @@ def generate_gpu(in_im_embs):
     im_emb = im_emb.detach().to('cpu').to(torch.float32)
     return output, im_emb
 
-@spaces.GPU()
+
 def generate(in_im_embs):
     output, im_emb = generate_gpu(in_im_embs)
     nsfw = maybe_nsfw(output.frames[0][len(output.frames[0])//2])