Update app.py

app.py

@@ -185,8 +185,8 @@ def generate_60(
     if use_resolution_binning:
         options["use_resolution_binning"] = True
     images = []
-    with torch.no_grad():
-      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:
@@ -235,8 +235,8 @@ def generate_90(
     if use_resolution_binning:
         options["use_resolution_binning"] = True
     images = []
-    with torch.no_grad():
-      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: