mirror of
https://github.com/vllm-project/vllm.git
synced 2025-10-20 14:53:52 +08:00
[Misc][Easy] Remove the space from the file name
This commit is contained in:
@ -765,7 +765,7 @@ def get_config_file_name(E: int,
|
||||
device_name = current_platform.get_device_name().replace(" ", "_")
|
||||
dtype_selector = "" if not dtype else f",dtype={dtype}"
|
||||
block_shape_selector = ("" if not block_shape or not all(block_shape) else
|
||||
f",block_shape={block_shape}")
|
||||
f",block_shape={block_shape}").replace(" ", "")
|
||||
return f"E={E},N={N},device_name={device_name}{dtype_selector}{block_shape_selector}.json" # noqa: E501
|
||||
|
||||
|
||||
|
@ -423,7 +423,7 @@ def get_w8a8_block_fp8_configs(N: int, K: int, block_n: int,
|
||||
# First look up if an optimized configuration is available in the configs
|
||||
# directory
|
||||
device_name = current_platform.get_device_name().replace(" ", "_")
|
||||
json_file_name = f"N={N},K={K},device_name={device_name},dtype=fp8_w8a8,block_shape=[{block_n}, {block_k}].json" # noqa: E501
|
||||
json_file_name = f"N={N},K={K},device_name={device_name},dtype=fp8_w8a8,block_shape=[{block_n},{block_k}].json" # noqa: E501
|
||||
|
||||
config_file_path = os.path.join(
|
||||
os.path.dirname(os.path.realpath(__file__)), "configs", json_file_name)
|
||||
|
Reference in New Issue
Block a user