Spaces:
Running
on
Zero
Running
on
Zero
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 = [
|