Hugging Face's logo

Spaces:
kxic
/
EscherNet
Sleeping

App Files Community
2
kxic commited on
Commit
94ff7fe
·
verified ·
1 Parent(s): 10ce721

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. gradio_demo/gradio_demo.py +1 -1
gradio_demo/gradio_demo.py CHANGED
@@ -18,7 +18,7 @@ import matplotlib.pyplot as pl
18
  pl.ion()
19
 
20
  CaPE_TYPE = "6DoF"
21
- device = 'cuda' if torch.cuda.is_available() else 'cpu'
22
  weight_dtype = torch.float16
23
  torch.backends.cuda.matmul.allow_tf32 = True # for gpu >= Ampere and pytorch >= 1.12
24
 
 
18
  pl.ion()
19
 
20
  CaPE_TYPE = "6DoF"
21
+ device = 'cuda' #if torch.cuda.is_available() else 'cpu'
22
  weight_dtype = torch.float16
23
  torch.backends.cuda.matmul.allow_tf32 = True # for gpu >= Ampere and pytorch >= 1.12
24