[1/2] Support deterministic inference with flashinfer attention backend (#10645)
Co-authored-by: hebiao064 <hebiaobuaa@gmail.com> Co-authored-by: Qiaolin-Yu <liin1211@outlook.com>
This commit is contained in:
@@ -75,6 +75,7 @@ class SamplingBatchInfo:
|
||||
@classmethod
|
||||
def from_schedule_batch(cls, batch: ScheduleBatch, vocab_size: int):
|
||||
global_server_args_dict = cls._get_global_server_args_dict()
|
||||
enable_deterministic = global_server_args_dict["enable_deterministic_inference"]
|
||||
|
||||
reqs = batch.reqs
|
||||
device = batch.device
|
||||
|
||||
Reference in New Issue
Block a user