Как убрать ошибку RuntimeError: Error(s) in loading state_dict for IntegratedAutoencoderKL