Spaces:
Sleeping
Sleeping
rynmurdock
commited on
Commit
•
5a8bcb9
1
Parent(s):
d8419d7
Update app.py
Browse files
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 |
-
|
465 |
def encode_space(x):
|
466 |
im_emb, _ = pipe.encode_image(
|
467 |
-
image,
|
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 |
|