Memory pool fix for upstream change about eagle (#4170)
This commit is contained in:
@@ -470,7 +470,7 @@ class MHATokenToKVPoolHost:
|
||||
def __init__(
|
||||
self,
|
||||
device_pool: MHATokenToKVPool,
|
||||
host_to_device_ratio: float = 2.0,
|
||||
host_to_device_ratio: float = 3.0,
|
||||
pin_memory: bool = False, # no need to use pin memory with the double buffering
|
||||
device: str = "cpu",
|
||||
):
|
||||
|
||||
Reference in New Issue
Block a user