Update top_p parameter in sampler function to 1.0 for enhanced sampling control in LTX2TextEncoder
This commit is contained in:
@@ -883,7 +883,7 @@ class LTX2TextEncoder(nn.Module):
|
|||||||
)
|
)
|
||||||
input_ids = mx.array(inputs["input_ids"])
|
input_ids = mx.array(inputs["input_ids"])
|
||||||
|
|
||||||
sampler = make_sampler(kwargs.get("temperature", 0.7), kwargs.get("top_p", 0.95), top_k=kwargs.get("top_k", -1))
|
sampler = make_sampler(kwargs.get("temperature", 0.7), kwargs.get("top_p", 1.0), top_k=kwargs.get("top_k", -1))
|
||||||
logits_processors = make_logits_processors(
|
logits_processors = make_logits_processors(
|
||||||
kwargs.get("logit_bias", None),
|
kwargs.get("logit_bias", None),
|
||||||
kwargs.get("repetition_penalty", 1.3),
|
kwargs.get("repetition_penalty", 1.3),
|
||||||
|
|||||||
Reference in New Issue
Block a user