Spaces:
Runtime error
Runtime error
Commit ·
b41bb43
1
Parent(s): 28244c5
polish
Browse files
app.py
CHANGED
|
@@ -56,8 +56,8 @@ pipe.to(device=DEVICE)
|
|
| 56 |
# TODO put back
|
| 57 |
@spaces.GPU
|
| 58 |
def compile_em():
|
| 59 |
-
pipe.unet = torch.compile(pipe.unet)
|
| 60 |
-
pipe.vae = torch.compile(pipe.vae)
|
| 61 |
autoencoder.model.forward = torch.compile(autoencoder.model.forward, backend='inductor', dynamic=True)
|
| 62 |
|
| 63 |
|
|
@@ -341,7 +341,7 @@ document.body.addEventListener('click', function(event) {
|
|
| 341 |
|
| 342 |
with gr.Blocks(css=css, head=js_head) as demo:
|
| 343 |
gr.Markdown('''### Zahir: Generative Recommenders for Unprompted, Scalable Exploration
|
| 344 |
-
Explore the latent space without
|
| 345 |
''', elem_id="description")
|
| 346 |
embs = gr.State([])
|
| 347 |
img_embs = gr.State([])
|
|
|
|
| 56 |
# TODO put back
|
| 57 |
@spaces.GPU
|
| 58 |
def compile_em():
|
| 59 |
+
pipe.unet = torch.compile(pipe.unet, mode='reduce-overhead')
|
| 60 |
+
pipe.vae = torch.compile(pipe.vae, mode='reduce-overhead')
|
| 61 |
autoencoder.model.forward = torch.compile(autoencoder.model.forward, backend='inductor', dynamic=True)
|
| 62 |
|
| 63 |
|
|
|
|
| 341 |
|
| 342 |
with gr.Blocks(css=css, head=js_head) as demo:
|
| 343 |
gr.Markdown('''### Zahir: Generative Recommenders for Unprompted, Scalable Exploration
|
| 344 |
+
Explore the latent space without prompting based on your feedback. Learn more on [the write-up](https://rynmurdock.github.io/posts/2024/3/generative_recomenders/).
|
| 345 |
''', elem_id="description")
|
| 346 |
embs = gr.State([])
|
| 347 |
img_embs = gr.State([])
|