schihei/llama2.c
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

docs typo

Browse Source
This commit is contained in:
Michael Cusack
2023-08-04 23:12:06 +07:00
parent f4c96b7339
commit 13f342af9e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
save_torchscript.py
+1 -1
View File
@@ -47,7 +47,7 @@ def main() -> None:
model_args = {k: globals()[k] for k in signature(ModelArgs).parameters} model_args = {k: globals()[k] for k in signature(ModelArgs).parameters}
model = Transformer(ModelArgs(**model_args)) model = Transformer(ModelArgs(**model_args))
# If requested zero params before saving the model. This is usful in # If requested zero params before saving the model. This is useful in
# conjunction with gzip_output. # conjunction with gzip_output.
if zero_params: if zero_params:
for p in model.parameters(): for p in model.parameters():