update space

app.py

@@ -74,6 +74,7 @@ def optimize_all(xm, models, initial_noise, noise_start_t, diffusion, latent_mod
 @spaces.GPU()
 @torch.no_grad()
 def generate_3d_with_shap_e(xm, diffusion, latent_model, device, prompt, rand_seed, state):
+    print("Check if I can use partial")
     set_seed(rand_seed)
     batch_size = 4
     guidance_scale = 15.0
@@ -416,6 +417,7 @@ def main():
                 inputs=[prompt, editing_choice[0], rand_seed],
                 outputs=[out_gen_1, out_gen_2, out_gen_3, out_gen_4, edited_1, edited_2, edited_3, edited_4],
                 fn=partial(optimize_all, xm, models, initial_noise, noise_start_t, diffusion, latent_model, device),
+                cache_examples=True,
             )