download_aim.py

from huggingface_hub import snapshot_download
import os

with open('/run/secrets/HF_TOKEN', 'r') as f:
    token = f.read().strip()

repo_id = os.environ['HF_REPO_ID']

snapshot_download(
    repo_id=repo_id,
    local_dir='.',
    allow_patterns='.aim/**',
    token=token
)