small change

app.py

@@ -34,7 +34,7 @@ torch.backends.cudnn.benchmark = True
 # -----------------------------
 # Model / pipeline loading
 # -----------------------------
-@torch.nograd()
+@torch.no_grad()
 @spaces.GPU
 def load_pipeline_single_gpu() -> FluxKontextSliderPipeline:
     pretrained = "black-forest-labs/FLUX.1-Kontext-dev"
@@ -295,6 +295,7 @@ def _encode_prompt(prompt: str):
 # Inference functions
 # -----------------------------
 @spaces.GPU
+@torch.no_grad()
 def generate_image_stack_edits(text_prompt, n_edits, input_image):
     """
     Compute n_edits images on a single GPU for slider values in (0,1],