drscotthawley commited on
Commit
00f0740
·
1 Parent(s): 04dd834

fixed zero gpu check

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -112,6 +112,7 @@ class Args:
112
  def process_image(image, repaint, busyness):
113
 
114
  print("CUDA check:")
 
115
  print(" zero.device = ",zero.device)
116
  accelerator = accelerate.Accelerator()
117
  device = accelerator.device
 
112
  def process_image(image, repaint, busyness):
113
 
114
  print("CUDA check:")
115
+ zero = torch.Tensor([0]).cuda()
116
  print(" zero.device = ",zero.device)
117
  accelerator = accelerate.Accelerator()
118
  device = accelerator.device