update replicate

replicate_predictor.py

@@ -27,23 +27,23 @@ class ReplicatePredictor(BasePredictor):
             le=8,
             default=4
         ),
-        log2_supercondition_factor: int = Input(
+        log2_temperature: float = Input(
-            description='Higher values result in better agreement with the text but a narrower variety of generated images',
+            description='A higher temperature results in more variety.',
-            ge=1,
+            ge=-3,
-            le=6,
+            le=3,
-            default=4
+            default=0
         ),
     ) -> Iterator[Path]:
         try: 
-            seed = -1
-            log2_mid_count = 3 if intermediate_outputs else 0
             image_stream = self.model.generate_image_stream(
-                text,
+                text = text,
-                seed,
+                seed = -1,
-                grid_size=grid_size,
+                grid_size = grid_size,
-                log2_mid_count=log2_mid_count,
+                log2_mid_count = 3 if intermediate_outputs else 0,
-                log2_supercondition_factor=log2_supercondition_factor,
+                temperature = 2 ** log2_temperature,
-                is_verbose=True
+                supercondition_factor = 2 ** 4,
+                top_k = 2 ** 8,
+                is_verbose = True
             )
 
             iter = 0