Spaces:
Sleeping
Sleeping
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([])
|