Fix is_all_ready for overlap copy (#1710)
This commit is contained in:
@@ -219,6 +219,7 @@ class SamplingBatchInfo:
|
|||||||
top_ps=self.top_ps,
|
top_ps=self.top_ps,
|
||||||
top_ks=self.top_ks,
|
top_ks=self.top_ks,
|
||||||
min_ps=self.min_ps,
|
min_ps=self.min_ps,
|
||||||
|
is_all_greedy=self.is_all_greedy,
|
||||||
need_min_p_sampling=self.need_min_p_sampling,
|
need_min_p_sampling=self.need_min_p_sampling,
|
||||||
vocab_size=self.vocab_size,
|
vocab_size=self.vocab_size,
|
||||||
device=self.device,
|
device=self.device,
|
||||||
|
|||||||
Reference in New Issue
Block a user