Spaces:

diffusers
/

benchmark

Paused

patrickvonplaten commited on Oct 11, 2023

Commit

2a3e2bb

1 Parent(s): dc0596c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ if USE_TORCH_COMPILE:
 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"

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