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

improve web ui

Browse Source
This commit is contained in:
hiyouga
2024-01-10 12:37:45 +08:00
parent 05ed4e8028
commit 1653c22438
4 changed files with 26 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -79,6 +79,10 @@ def get_current_device() -> torch.device:
return torch.device(device)
def get_device_count() -> int:
return torch.cuda.device_count()
def get_logits_processor() -> "LogitsProcessorList":
r"""
Gets logits processor that removes NaN and Inf logits.