.venv/
model-checkpoints/**
__pycache__/
*.ipynb