valknar/audiocraft-ui
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix GPUMemoryManager method call - use get_status()

Browse Source 
🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
Sebastian Krüger
2025-11-26 23:42:29 +01:00
parent dcf369db13
commit e53314023c
1 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/ui/app.py
View File

@@ -75,13 +75,11 @@ class AudioCraftApp:
def _get_gpu_status(self) -> dict[str, Any]: def _get_gpu_status(self) -> dict[str, Any]:
"""Get GPU memory status.""" """Get GPU memory status."""
if self.gpu_manager: if self.gpu_manager:
return { try:
"used_gb": self.gpu_manager.get_used_memory() / 1024**3, return self.gpu_manager.get_status()
"total_gb": self.gpu_manager.total_memory / 1024**3, except Exception:
"utilization_percent": self.gpu_manager.get_utilization(), pass
"available_gb": self.gpu_manager.get_available_memory() / 1024**3, return {"total_gb": 24, "used_gb": 0, "free_gb": 24, "utilization_percent": 0, "available_for_models_gb": 14}
}
return {"used_gb": 0, "total_gb": 24, "utilization_percent": 0, "available_gb": 24}
async def _generate(self, **kwargs) -> tuple[Any, Any]: async def _generate(self, **kwargs) -> tuple[Any, Any]:
"""Generate audio using the generation service.""" """Generate audio using the generation service."""