GradientGuru's picture
fall back to self.config for return_dict
ff1fbc5