423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-16 03:40:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

add API demo from #1

Browse Source
This commit is contained in:
hiyouga
2023-06-05 21:32:18 +08:00
parent 06e1b120e1
commit 3d8d5ee5d5
5 changed files with 191 additions and 178 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/utils/other.py
View File

@@ -142,7 +142,7 @@ def load_valuehead_params(model: torch.nn.Module, checkpoint_dir: os.PathLike) -
def smooth(scalars: List[float], weight: Optional[float] = 0.9) -> List[float]:
"""
r"""
EMA implementation according to TensorBoard.
"""
last = scalars[0]