Update app.py

app.py

@@ -3,7 +3,7 @@ import torch
 from diffusers import AutoPipelineForText2Image
 import time
 
-USE_TORCH_COMPILE = False
+USE_TORCH_COMPILE = True
 
 dtype = torch.float16
 device = torch.device("cuda:0")
@@ -16,6 +16,7 @@ if USE_TORCH_COMPILE:
     pipeline.unet = torch.compile(pipeline.unet, mode="reduce-overhead", fullgraph=True)
     
 def generate(num_images_per_prompt: int = 1):
+    print("Version", torch.__version__)
     prompt = 77 * "a"
     num_inference_steps = 40
     start_time = time.time()