Update app.py

app.py

@@ -28,13 +28,13 @@ def main(image, prompt, percent_noise): # percent_noise = 0.5, 0.02, 0.999
   z = pipe.scheduler.add_noise(clean_z, noise, timestep).half()
 
   # Run the diffusion model
-  with autocast('cuda'):
-    # The 'num_inference_steps=100' arguments means that, if you were running
-    # the full diffusion model (i.e., percent_noise = 0.999), it would be sampling
-    # at 2x frequency compared to the normal stable diffusion model (which uses 50 steps).
-    # This way, running percent_noise = 0.5 yields 50 inference steps,
-    # and running percent_noise = 0.2 yields 20 inference steps, etc.
-    return pipe(prompt=prompt, latents=z, num_inference_steps=100, percent_noise=percent_noise).images[0]
+  #with autocast('cuda'):
+  # The 'num_inference_steps=100' arguments means that, if you were running
+  # the full diffusion model (i.e., percent_noise = 0.999), it would be sampling
+  # at 2x frequency compared to the normal stable diffusion model (which uses 50 steps).
+  # This way, running percent_noise = 0.5 yields 50 inference steps,
+  # and running percent_noise = 0.2 yields 20 inference steps, etc.
+  return pipe(prompt=prompt, latents=z, num_inference_steps=100, percent_noise=percent_noise).images[0]