Files
ai-toolkit/testing/test_model_loading.py