Difussion models support code