Skip to content

Commit 85e1463

Browse files
committed
use fake mem info when cuda not available
1 parent 4df5cef commit 85e1463

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

src/optimization/memory_manager.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,15 @@
2323
def get_basic_vram_info():
2424
"""🔍 Méthode basique avec PyTorch natif"""
2525
if not torch.cuda.is_available():
26-
return {"error": "CUDA not available"}
26+
from loguru import logger
27+
logger.info("SeedVR2_VideoUpscaler: use fake memory info")
28+
free_memory, total_memory = (24893718528*2, 25438126080*2)
29+
free_gb = free_memory / (1024**3)
30+
total_gb = total_memory / (1024**3)
31+
return {
32+
"free_gb": free_gb,
33+
"total_gb": total_gb
34+
}
2735

2836
# Mémoire libre et totale (en bytes)
2937
free_memory, total_memory = torch.cuda.mem_get_info()

0 commit comments

Comments
 (0)