fix: resolve top_k shadowing and None-guard bug in sample_from_logits (PR shiyu-coder#238)

Author: aidan46

model/kronos.py

@@ -372,9 +372,8 @@ def top_k_top_p_filtering(
 
 def sample_from_logits(logits, temperature=1.0, top_k=None, top_p=None, sample_logits=True):
     logits = logits / temperature
-    if top_k is not None or top_p is not None:
-        if top_k > 0 or top_p < 1.0:
-            logits = top_k_top_p_filtering(logits, top_k=top_k, top_p=top_p)
+    if (top_k is not None and top_k > 0) or (top_p is not None and top_p < 1.0):
+        logits = top_k_top_p_filtering(logits, top_k=top_k or 0, top_p=top_p or 1.0)
 
     probs = F.softmax(logits, dim=-1)