Hugging Face's logo

Spaces:
cocktailpeanut
/
ZeST
Runtime error

App Files Community
cocktailpeanut commited on
Commit
fbfeead
·
1 Parent(s): bca2d24

update

Browse files
Files changed (1) hide show
  1. demo_gradio.py +1 -1
demo_gradio.py CHANGED
@@ -46,7 +46,7 @@ ip_ckpt = "sdxl_models/ip-adapter_sdxl_vit-h.bin"
46
  controlnet_path = "diffusers/controlnet-depth-sdxl-1.0"
47
  #device = "cuda"
48
  #torch.cuda.empty_cache()
49
- device = devicetorch.device(torch)
50
  devicetorch.empty_cache(torch)
51
 
52
  # load SDXL pipeline
 
46
  controlnet_path = "diffusers/controlnet-depth-sdxl-1.0"
47
  #device = "cuda"
48
  #torch.cuda.empty_cache()
49
+ device = devicetorch.get(torch)
50
  devicetorch.empty_cache(torch)
51
 
52
  # load SDXL pipeline