We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 4df5cef commit 85e1463Copy full SHA for 85e1463
src/optimization/memory_manager.py
@@ -23,7 +23,15 @@
23
def get_basic_vram_info():
24
"""🔍 Méthode basique avec PyTorch natif"""
25
if not torch.cuda.is_available():
26
- return {"error": "CUDA not available"}
+ 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
+ }
35
36
# Mémoire libre et totale (en bytes)
37
free_memory, total_memory = torch.cuda.mem_get_info()
0 commit comments