| |
| """Bootstrap: downloads source from HF repo and runs the full build.""" |
| import subprocess, sys, os |
| subprocess.check_call([sys.executable, "-m", "pip", "install", "-q", |
| "huggingface_hub", "jax", "jaxlib", "flax", "optax", "numpy"]) |
| from huggingface_hub import hf_hub_download |
| repo = "gamer729/cognithread-ssm" |
| files = ["cognithread_full.py"] |
| for f in files: |
| path = hf_hub_download(repo_id=repo, filename=f, repo_type="model") |
| os.system(f"cp {path} /app/{f}") |
| print(f"Downloaded {f}") |
| os.system(f"{sys.executable} /app/cognithread_full.py") |
|
|
