diff --git a/examples/mixtral.py b/examples/mixtral.py index 2dd226aa33..39f6d19af5 100644 --- a/examples/mixtral.py +++ b/examples/mixtral.py @@ -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")