github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-10 21:40:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

[hotfix] fix typo s/get_defualt_parser /get_default_parser (#5548)

Browse Source
This commit is contained in:
digger yu
2024-04-07 19:04:58 +08:00
committed by GitHub
parent a799ca343b
commit 341263df48
3 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/language/grok-1/inference.py
View File

@@ -2,10 +2,10 @@ import time
import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
from utils import get_defualt_parser, inference, print_output
from utils import get_default_parser, inference, print_output
if __name__ == "__main__":
parser = get_defualt_parser()
parser = get_default_parser()
args = parser.parse_args()
start = time.time()
torch.set_default_dtype(torch.bfloat16)