Update app.py

app.py

@@ -149,7 +149,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     return seed
-
+    
 @spaces.GPU(duration=60)
 def generate(
     model_choice: str,
@@ -189,7 +189,8 @@ def generate(
         options["use_resolution_binning"] = True
 
     images = []
-    for i in range(0, num_images, BATCH_SIZE):
+    with torch.no_grad():
+      for i in range(0, num_images, BATCH_SIZE):
         batch_options = options.copy()
         batch_options["prompt"] = options["prompt"][i:i+BATCH_SIZE]
         if "negative_prompt" in batch_options: