Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -16,6 +16,11 @@ from nodes import NODE_CLASS_MAPPINGS
|
|
16 |
from comfy import model_management
|
17 |
import folder_paths
|
18 |
|
|
|
|
|
|
|
|
|
|
|
19 |
# Diret贸rio base e de sa铆da
|
20 |
BASE_DIR = os.path.dirname(os.path.realpath(__file__))
|
21 |
output_dir = os.path.join(BASE_DIR, "output")
|
|
|
16 |
from comfy import model_management
|
17 |
import folder_paths
|
18 |
|
19 |
+
print("CUDA dispon铆vel:", torch.cuda.is_available())
|
20 |
+
print("Quantidade de GPUs:", torch.cuda.device_count())
|
21 |
+
if torch.cuda.is_available():
|
22 |
+
print("GPU atual:", torch.cuda.get_device_name(0))
|
23 |
+
|
24 |
# Diret贸rio base e de sa铆da
|
25 |
BASE_DIR = os.path.dirname(os.path.realpath(__file__))
|
26 |
output_dir = os.path.join(BASE_DIR, "output")
|