Convert FLASHINFER_WORKSPACE_SIZE to integer (#10731)
This commit is contained in:
@@ -37,8 +37,8 @@ class GlobalConfig:
|
|||||||
)
|
)
|
||||||
# Runtime constants: others
|
# Runtime constants: others
|
||||||
self.retract_decode_steps = 20
|
self.retract_decode_steps = 20
|
||||||
self.flashinfer_workspace_size = os.environ.get(
|
self.flashinfer_workspace_size = int(
|
||||||
"FLASHINFER_WORKSPACE_SIZE", 384 * 1024 * 1024
|
os.environ.get("FLASHINFER_WORKSPACE_SIZE", 384 * 1024 * 1024)
|
||||||
)
|
)
|
||||||
|
|
||||||
# Output tokenization configs
|
# Output tokenization configs
|
||||||
|
|||||||
Reference in New Issue
Block a user