update

inference.py

@@ -54,9 +54,9 @@ def inference_fn(
         pipe = StableDiffusionPipeline.from_pretrained(os.path.join('experiments', model_id),torch_dtype=torch.float16).to('cuda')
     else:
         pipe = StableDiffusionPipeline.from_pretrained(os.path.join('experiments', model_id)).to('cpu')
-    # make directory to save images
-    image_root_folder = os.path.join('experiments', model_id, 'inference')
-    os.makedirs(image_root_folder, exist_ok = True)
+    # # make directory to save images
+    # image_root_folder = os.path.join('experiments', model_id, 'inference')
+    # os.makedirs(image_root_folder, exist_ok = True)
 
     # if prompt is None and args.template_name is None:
     #     raise ValueError("please input a single prompt through'--prompt' or select a batch of prompts using '--template_name'.")
@@ -77,22 +77,22 @@ def inference_fn(
         prompt = prompt.lower().replace("<r>", "<R>").format("<R>")
 
 
-        # make sub-folder
-        image_folder = os.path.join(image_root_folder, prompt, 'samples')
-        os.makedirs(image_folder, exist_ok = True)
+        # # make sub-folder
+        # image_folder = os.path.join(image_root_folder, prompt, 'samples')
+        # os.makedirs(image_folder, exist_ok = True)
 
         # batch generation
         images = pipe(prompt, num_inference_steps=ddim_steps, guidance_scale=guidance_scale, num_images_per_prompt=num_samples).images
 
-        # save generated images
-        for idx, image in enumerate(images):
-            image_name = f"{str(idx).zfill(4)}.png"
-            image_path = os.path.join(image_folder, image_name)
-            image.save(image_path)
+        # # save generated images
+        # for idx, image in enumerate(images):
+        #     image_name = f"{str(idx).zfill(4)}.png"
+        #     image_path = os.path.join(image_folder, image_name)
+        #     image.save(image_path)
 
         # save a grid of images
         image_grid = make_image_grid(images, rows=2, cols=math.ceil(num_samples/2))
-        image_grid_path = os.path.join(image_root_folder, prompt, f'{prompt}.png')
+        # image_grid_path = os.path.join(image_root_folder, prompt, f'{prompt}.png')
 
         return image_grid