| 24 | |
| 25 | |
| 26 | def make_args_parser(): |
| 27 | parser = argparse.ArgumentParser( |
| 28 | "MeshXL: Neural Coordinate Field for Generative 3D Foundation Models", |
| 29 | add_help=False |
| 30 | ) |
| 31 | ##### model config ##### |
| 32 | parser.add_argument("--llm", default='mesh-xl/mesh-xl-350m', type=str) |
| 33 | parser.add_argument("--n_discrete_size", default=128, type=int) |
| 34 | parser.add_argument("--text_condition", default='openai/clip-vit-base-patch32', type=str) |
| 35 | parser.add_argument("--test_ckpt", default='mesh-xl/x-mesh-xl-350m/pytorch_model.bin', type=str) |
| 36 | parser.add_argument("--text", default='3d model of a chair', type=str) |
| 37 | parser.add_argument("--output_dir", default='outputs', type=str) |
| 38 | parser.add_argument("--num_samples", default=4, type=int) |
| 39 | parser.add_argument("--top_k", default=50, type=int) |
| 40 | parser.add_argument("--top_p", default=0.95, type=float) |
| 41 | parser.add_argument("--temperature", default=0.1, type=float) |
| 42 | |
| 43 | args = parser.parse_args() |
| 44 | return args |
| 45 | |
| 46 | |
| 47 | if __name__ == '__main__': |