diff --git a/swift/pipelines/sampling/distill_sampler.py b/swift/pipelines/sampling/distill_sampler.py index 825753edb0..61d2c6e9d3 100644 --- a/swift/pipelines/sampling/distill_sampler.py +++ b/swift/pipelines/sampling/distill_sampler.py @@ -32,7 +32,7 @@ def infer( for infer_request in infer_requests: completion = self.client.chat.completions.create( model=self.model, - messages=infer_request['messages'], + messages=infer_request.messages, temperature=request_config.temperature, top_p=request_config.top_p, max_tokens=request_config.max_tokens,