Files
ai-toolkit/toolkit/models/diffusion_feature_extraction.py