Spaces:

rynmurdock
/

Blue_Tigers

Runtime error

rynmurdock commited on May 19, 2024

Commit

0d0b2b8

verified ·

1 Parent(s): 34e28a7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -125,7 +125,7 @@ pipe.to(device=DEVICE)
 #leave_im_emb.detach().to('cpu')
-@spaces.GPU()
 def generate(in_im_embs):
     in_im_embs = in_im_embs.to('cuda').unsqueeze(0).unsqueeze(0)
     #im_embs = torch.cat((torch.zeros(1, 1280, device=DEVICE, dtype=dtype), in_im_embs), 0)

 #leave_im_emb.detach().to('cpu')
+@spaces.GPU(duration=10)
 def generate(in_im_embs):
     in_im_embs = in_im_embs.to('cuda').unsqueeze(0).unsqueeze(0)
     #im_embs = torch.cat((torch.zeros(1, 1280, device=DEVICE, dtype=dtype), in_im_embs), 0)