Model loading and model storage: https://github.com/intractai/DockerAgent/blob/main/src/modeling.py For generation and inference: https://github.com/intractai/DockerAgent/tree/main/src/routers