tinygrad/tinygrad
2
0
Fork 3
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-08-29 15:16:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix help message (#2705)

Browse Source 
llama -> mixtral
This commit is contained in:
Ivan Vnučec
2023-12-10 22:04:35 -08:00
committed by GitHub
parent 59ab3675a3
commit 8d206f6bfd
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/mixtral.py
+1 -1
View File
@@ -21,7 +21,7 @@ class MixtureFeedForward:
return ret
if __name__ == "__main__":
parser = argparse.ArgumentParser(description="Run LLaMA in tinygrad", formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser = argparse.ArgumentParser(description="Run Mixtral in tinygrad", formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument("--count", type=int, default=30, help="Max number of tokens to generate")
parser.add_argument("--temperature", type=float, default=0.7, help="Temperature in the softmax")
parser.add_argument("--timing", action="store_true", help="Print timing per token")