diff --git a/ginkgo_ai_client/queries.py b/ginkgo_ai_client/queries.py index e89960a..87705f3 100644 --- a/ginkgo_ai_client/queries.py +++ b/ginkgo_ai_client/queries.py @@ -502,13 +502,13 @@ def to_request_params(self) -> Dict: "species": self.species, "temperature": self.temperature, "decoding_order_strategy": self.decoding_order_strategy, - "num_to_decode_per_step": self.unmaskings_per_step, + "unmaskings_per_step": self.unmaskings_per_step, "num_samples": self.num_samples, } return { "model": self.model, "text": json.dumps(data), - "transforms": [{"type": "GENERATE"}], + "transforms": [{"type": "MRNA_DIFFUSION_GENERATE"}], } def parse_response(self, results: Dict) -> MultimodalDiffusionMaskedResponse: