rynmurdock commited on
Commit
b41bb43
1 Parent(s): 28244c5
Files changed (1) hide show
  1. app.py +3 -3
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 text prompts, based on your preferences. 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([])
 
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([])