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

Typo

Browse Source
This commit is contained in:
Michael Cusack
2023-08-04 17:22:27 +07:00
parent f8d45f180d
commit 9f8e0857ee
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
save_model.py
+1 -1
View File
@@ -2,7 +2,7 @@
#!/usr/bin/env python #!/usr/bin/env python
"""Saves the model as a TorchScript. """Saves the model as a TorchScript.
The resulting file can be loaded in C++ code and then used for training or infrence with: The resulting file can be loaded in C++ code and then used for training or inference with:
#include <torch/script.h> #include <torch/script.h>
torch::jit::Module module = torch::jit::load("model.pt") torch::jit::Module module = torch::jit::load("model.pt")
""" """