Update launch.py

2023-07-25 08:47:05 +05:30 · 2023-07-25 08:47:05 +05:30 · 4e06dad9f6
parent a0a220d84b
commit 4e06dad9f6
1 changed files with 0 additions and 5 deletions
--- a/projects/gradio_demo/launch.py
+++ b/projects/gradio_demo/launch.py
@ -59,11 +59,6 @@ class InferencerCache:
        if len(cls._cache) == cls.max_size:
            cls._cache.pop(cls.max_size - 1)
            torch.cuda.empty_cache()
-            try:
-                import torch.mps
-                torch.mps.empty_cache()
-            except ImportError:
-                pass
        device = get_free_device()
        instance = callback(device=device)
        logger.info(f'New instance {instance_name} on {device}.')