Spaces:
Running
Running
default to cuda for zero space
Browse files
app.py
CHANGED
|
@@ -10,7 +10,8 @@ from game_of_life import GameOfLife
|
|
| 10 |
from utils import resize_image, generate_image_from_grid
|
| 11 |
|
| 12 |
|
| 13 |
-
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
|
| 14 |
controlnet = QRControlNet(device=device)
|
| 15 |
|
| 16 |
|
|
|
|
| 10 |
from utils import resize_image, generate_image_from_grid
|
| 11 |
|
| 12 |
|
| 13 |
+
device = "cuda" # if torch.cuda.is_available() else "cpu"
|
| 14 |
+
print(f"Using {device=})
|
| 15 |
controlnet = QRControlNet(device=device)
|
| 16 |
|
| 17 |
|