423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-15 19:30:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix #1558

Browse Source
This commit is contained in:
hiyouga
2023-11-19 14:15:47 +08:00
parent ff6056405d
commit 1740131d63
8 changed files with 35 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/llmtuner/extras/misc.py
View File

@@ -66,8 +66,7 @@ def count_parameters(model: torch.nn.Module) -> Tuple[int, int]:
def get_current_device() -> str:
import accelerate
from accelerate import Accelerator
dummy_accelerator = Accelerator()
dummy_accelerator = accelerate.Accelerator()
if accelerate.utils.is_xpu_available():
return "xpu:{}".format(dummy_accelerator.local_process_index)
else: