feat(core): APP use new SDK component (#1050)

This commit is contained in:
Fangyin Cheng
2024-01-10 10:39:04 +08:00
committed by GitHub
parent e11b72c724
commit fa8b5b190c
242 changed files with 2768 additions and 2163 deletions

View File

@@ -1,6 +1,6 @@
from typing import List, Tuple
from dataclasses import dataclass
import logging
from dataclasses import dataclass
from typing import List, Tuple
logger = logging.getLogger(__name__)
@@ -17,9 +17,10 @@ def _clear_model_cache(device="cuda"):
def _clear_torch_cache(device="cuda"):
import torch
import gc
import torch
gc.collect()
if device != "cpu":
if torch.has_mps: