Spaces:
Runtime error
Runtime error
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")
|