init

app.py

@@ -82,7 +82,7 @@ class Pipeline:
     @spaces.GPU
     @torch.no_grad()
     @torch.autocast(device_type="cuda", dtype=torch.bfloat16)
-    def __call__(self, y, num_images, seed, image_height, image_width, num_steps, guidance, timeshift, order):
+    def __call__(self, y, seed, image_height, image_width, num_steps, guidance, timeshift, order):
         diffusion_sampler = AdamLMSampler(
             order=order,
             scheduler=LinearScheduler(),
@@ -96,10 +96,10 @@ class Pipeline:
         image_width = image_width // 32 * 32
         self.denoiser.decoder_patch_scaling_h = image_height / 512
         self.denoiser.decoder_patch_scaling_w = image_width / 512
-        xT = torch.randn((num_images, 3, image_height, image_width), device="cpu", dtype=torch.float32,
+        xT = torch.randn((1, 3, image_height, image_width), device="cpu", dtype=torch.float32,
                          generator=generator).cuda()
         with torch.no_grad():
-            condition, uncondition = conditioner([y,]*num_images)
+            condition, uncondition = conditioner([y,]*1)
 
 
         # Sample images:
@@ -182,7 +182,6 @@ if __name__ == "__main__":
                 guidance = gr.Slider(minimum=0.1, maximum=10.0, step=0.1, label="CFG", value=4.0)
                 image_height = gr.Slider(minimum=128, maximum=1024, step=32, label="image height", value=512)
                 image_width = gr.Slider(minimum=128, maximum=1024, step=32, label="image width", value=512)
-                num_images = gr.Slider(minimum=1, maximum=4, step=1, label="num images", value=4)
                 label = gr.Textbox(label="positive prompt", value="a photo of a cat")
                 seed = gr.Slider(minimum=0, maximum=1000000, step=1, label="seed", value=0)
                 timeshift = gr.Slider(minimum=0.1, maximum=5.0, step=0.1, label="timeshift", value=3.0)
@@ -196,7 +195,6 @@ if __name__ == "__main__":
         btn.click(fn=pipeline,
                   inputs=[
                       label,
-                      num_images,
                       seed,
                       image_height,
                       image_width,