movimento / __init__.py
rydlrKE's picture
Fix CUDA import order - import spaces before torch (commit e28bffd)
2a5255e verified
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0
# ruff: noqa: I001
import argparse
from kimodo.model import DEFAULT_MODEL
from kimodo.model.registry import resolve_model_name
from .app import Demo
def main() -> None:
parser = argparse.ArgumentParser(description="Run the kimodo demo UI.")
parser.add_argument(
"--model",
type=str,
default=DEFAULT_MODEL,
help="Default model to load (e.g. Kimodo-SOMA-RP-v1, kimodo-soma-rp, or SOMA).",
)
args = parser.parse_args()
resolved = resolve_model_name(args.model, "Kimodo")
demo = Demo(default_model_name=resolved)
demo.run()
if __name__ == "__main__":
main()