Spaces:

diffusers
/

benchmark

Paused

patrickvonplaten commited on Oct 11, 2023

Commit

dc0596c

1 Parent(s): 8996688

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,8 @@ if USE_TORCH_COMPILE:
 def generate(num_images_per_prompt: int = 1):
     print("Start...")
-    print("Version", torch.__version__)
     for _ in range(3):
         prompt = 77 * "a"

 def generate(num_images_per_prompt: int = 1):
     print("Start...")
+    print("Torch version", torch.__version__)
+    print("Torch CUDA version", torch.cuda.version)
     for _ in range(3):
         prompt = 77 * "a"