fix default top-k

cube/cube3d/inference/engine.py

@@ -266,7 +266,7 @@ class Engine:
         guidance_scale: float = 3.0,
         resolution_base: float = 8.0,
         chunk_size: int = 100_000,
-        top_k: int = 5,
+        top_k: int = 1,
     ):
         """
         Generates a 3D mesh from text prompts using a GPT model and shape decoder.

cube/cube3d/inference/logits_postprocesses.py

@@ -10,7 +10,6 @@ def top_k_filtering(logits, top_k: int = 1):
     Args:
         logits: A tensor of logits to be filtered. Expected shape is [..., vocab_size].
         top_k: If > 0, only keep the top k tokens with highest probability.
-        top_p: If < 1.0, only keep tokens whose cumulative probability is below this threshold.
 
     Returns:
         A tensor of logits where values outside the top-k/top-p threshold are set to -∞.