Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -22,6 +22,16 @@ output_dir = os.path.join(BASE_DIR, "output")
|
|
| 22 |
os.makedirs(output_dir, exist_ok=True)
|
| 23 |
folder_paths.set_output_directory(output_dir)
|
| 24 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 25 |
# 4. Download de Modelos
|
| 26 |
def download_models():
|
| 27 |
models = [
|
|
|
|
| 22 |
os.makedirs(output_dir, exist_ok=True)
|
| 23 |
folder_paths.set_output_directory(output_dir)
|
| 24 |
|
| 25 |
+
# 4. Diagn贸stico CUDA
|
| 26 |
+
print("Python version:", sys.version)
|
| 27 |
+
print("Torch version:", torch.__version__)
|
| 28 |
+
print("CUDA dispon铆vel:", torch.cuda.is_available())
|
| 29 |
+
print("Quantidade de GPUs:", torch.cuda.device_count())
|
| 30 |
+
if torch.cuda.is_available():
|
| 31 |
+
print("GPU atual:", torch.cuda.get_device_name(0))
|
| 32 |
+
else:
|
| 33 |
+
print("GPU n茫o dispon铆vel. Usando CPU.")
|
| 34 |
+
|
| 35 |
# 4. Download de Modelos
|
| 36 |
def download_models():
|
| 37 |
models = [
|