tweaks to mrna_foundation diffusion parameters

ginkgobioworks · Jan 10, 2025 · 81bc0ee · 81bc0ee
1 parent 212b842
commit 81bc0ee
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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: