fix bug

modules/base_interface.py

@@ -9,8 +9,9 @@ class BaseInterface:
 
     @staticmethod
     def release_cuda_memory():
-        torch.cuda.empty_cache()
-        torch.cuda.reset_max_memory_allocated()
+        if torch.cuda.is_available():
+            torch.cuda.empty_cache()
+            torch.cuda.reset_max_memory_allocated()
 
     @staticmethod
     def remove_input_files(file_paths: List[str]):