github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-05 02:20:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: can not answer on mac m1-> mps device (#85)

Browse Source 
fix: #86
This commit is contained in:
magic.chen 2023-05-24 13:33:35 +08:00 committed by GitHub
commit 562d5a98cc
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pilot/model/loader.py
View File

@ -9,6 +9,7 @@ from typing import Optional
from pilot.model.compression import compress_module from pilot.model.compression import compress_module
from pilot.model.adapter import get_llm_model_adapter from pilot.model.adapter import get_llm_model_adapter
from pilot.utils import get_gpu_memory from pilot.utils import get_gpu_memory
from pilot.configs.model_config import DEVICE
from pilot.model.llm.monkey_patch import replace_llama_attn_with_non_inplace_operations from pilot.model.llm.monkey_patch import replace_llama_attn_with_non_inplace_operations
def raise_warning_for_incompatible_cpu_offloading_configuration( def raise_warning_for_incompatible_cpu_offloading_configuration(
@ -50,7 +51,7 @@ class ModelLoader(metaclass=Singleton):
def __init__(self, def __init__(self,
model_path) -> None: model_path) -> None:
self.device = "cuda" if torch.cuda.is_available() else "cpu" self.device = DEVICE
self.model_path = model_path self.model_path = model_path
self.kwargs = { self.kwargs = {
"torch_dtype": torch.float16, "torch_dtype": torch.float16,