mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-10-23 18:31:22 +00:00
feat(core): APP use new SDK component (#1050)
This commit is contained in:
@@ -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:
|
||||
|
Reference in New Issue
Block a user