Training flux-schnell model - Runpod