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

add API demo from #1

Browse Source 
Former-commit-id: c955edcef168da44257c5b50d7bc59266d909782
This commit is contained in:
hiyouga
2023-06-05 21:32:18 +08:00
parent a441a411cb
commit 5bedf2b21a
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]