dginovker commited on
Commit
6082b54
·
verified ·
1 Parent(s): e2cff89

Verify zero support

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -10,6 +10,9 @@ import os
10
  from PIL import Image
11
  import spaces
12
 
 
 
 
13
  from diffusers import (
14
  StableDiffusionPipeline,
15
  StableDiffusionControlNetImg2ImgPipeline,
@@ -324,3 +327,4 @@ model: https://huggingface.co/DionTimmer/controlnet_qrcode-control_v1p_sd15
324
  os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
325
  blocks.queue(max_size=20,api_open=False)
326
  blocks.launch(share=bool(os.environ.get("SHARE", False)), show_api=False)
 
 
10
  from PIL import Image
11
  import spaces
12
 
13
+ print(f"Is CUDA available: {torch.cuda.is_available()}")
14
+ print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
15
+
16
  from diffusers import (
17
  StableDiffusionPipeline,
18
  StableDiffusionControlNetImg2ImgPipeline,
 
327
  os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
328
  blocks.queue(max_size=20,api_open=False)
329
  blocks.launch(share=bool(os.environ.get("SHARE", False)), show_api=False)
330
+