Update app.py

app.py

@@ -271,9 +271,9 @@ def edit_inference(net, prompt, negative_prompt, guidance_scale, ddim_steps, see
         elif t<=start_noise:
             network.proj = torch.nn.Parameter(edited_weights)
             network.reset()
-          
-        with network:
-            noise_pred = unet(latent_model_input, t, encoder_hidden_states=text_embeddings, timestep_cond= None).sample
+        with torch.no_grad():
+            with network:
+                noise_pred = unet(latent_model_input, t, encoder_hidden_states=text_embeddings, timestep_cond= None).sample
                     
                 
         #guidance