Update app.py

app.py

@@ -24,7 +24,7 @@ CACHE_EXAMPLES = False #torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES"
 MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "1536"))
 USE_TORCH_COMPILE = False
 ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD") == "1"
-PREVIEW_IMAGES = True
+PREVIEW_IMAGES = False
 
 dtype = torch.bfloat16
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
@@ -86,7 +86,7 @@ def generate(
     profile: gr.OAuthProfile | None = None,
 ) -> PIL.Image.Image:
     
-    previewer.eval().requires_grad_(False).to(device).to(dtype)
+    #previewer.eval().requires_grad_(False).to(device).to(dtype)
     prior_pipeline.to(device)
     decoder_pipeline.to(device)
     
@@ -102,10 +102,9 @@ def generate(
         guidance_scale=prior_guidance_scale,
         num_images_per_prompt=num_images_per_prompt,
         generator=generator,
-        callback_on_step_end=callback_prior,
-        callback_on_step_end_tensor_inputs=['latents']
+        #callback_on_step_end=callback_prior,
+        #callback_on_step_end_tensor_inputs=['latents']
     )
-    print(prior_output)
     if PREVIEW_IMAGES:
         for _ in range(len(DEFAULT_STAGE_C_TIMESTEPS)):
             r = next(prior_output)