Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
00f0740
1
Parent(s):
04dd834
fixed zero gpu check
Browse files
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
|