Spaces:
Runtime error
Runtime error
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 |
|