()
| 62 | self.ffn_dim = 64 |
| 63 | |
| 64 | def parse_args(): |
| 65 | parser = argparse.ArgumentParser(description="Prefill on simulator") |
| 66 | parser.add_argument("--config", default="config.json", type=str, help="Config file") |
| 67 | args = parser.parse_args() |
| 68 | return args |
| 69 | |
| 70 | def main(): |
| 71 | args = parse_args() |