rynmurdock commited on
Commit
49ba956
1 Parent(s): 529de1b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -461,10 +461,10 @@ scheduler = BackgroundScheduler()
461
  scheduler.add_job(func=background_next_image, trigger="interval", seconds=1)
462
  scheduler.start()
463
 
464
- @spaces.GPU()
465
  def encode_space(x):
466
  im_emb, _ = pipe.encode_image(
467
- image, DEVICE, 1, output_hidden_state
468
  )
469
  return im_emb.detach().to('cpu').to(torch.float32)
470
 
 
461
  scheduler.add_job(func=background_next_image, trigger="interval", seconds=1)
462
  scheduler.start()
463
 
464
+
465
  def encode_space(x):
466
  im_emb, _ = pipe.encode_image(
467
+ image, 'cpu', 1, output_hidden_state
468
  )
469
  return im_emb.detach().to('cpu').to(torch.float32)
470