| |
| import argparse |
| from huggingface_hub import hf_hub_download |
| from coe.light.model.main import LitORionModelOptimized |
|
|
|
|
| def main(): |
| parser = argparse.ArgumentParser() |
| parser.add_argument("--repo_id", required=True, help="e.g. your-name/mnemodyn-gordon333") |
| parser.add_argument("--revision", default="main") |
| args = parser.parse_args() |
|
|
| ckpt_path = hf_hub_download(repo_id=args.repo_id, filename="Orion_333/model.ckpt", revision=args.revision) |
| model = LitORionModelOptimized.load_from_checkpoint(ckpt_path, map_location="cpu") |
| model.eval() |
| print("Loaded model from", ckpt_path) |
|
|
|
|
| if __name__ == "__main__": |
| main() |
|
|
