mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-06-08 04:22:15 +08:00
[Misc][Easy] Remove the space from the file name
This commit is contained in:
parent
76abd0c881
commit
9cdea30b4f
@ -765,7 +765,7 @@ def get_config_file_name(E: int,
|
|||||||
device_name = current_platform.get_device_name().replace(" ", "_")
|
device_name = current_platform.get_device_name().replace(" ", "_")
|
||||||
dtype_selector = "" if not dtype else f",dtype={dtype}"
|
dtype_selector = "" if not dtype else f",dtype={dtype}"
|
||||||
block_shape_selector = ("" if not block_shape or not all(block_shape) else
|
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
|
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
|
# First look up if an optimized configuration is available in the configs
|
||||||
# directory
|
# directory
|
||||||
device_name = current_platform.get_device_name().replace(" ", "_")
|
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(
|
config_file_path = os.path.join(
|
||||||
os.path.dirname(os.path.realpath(__file__)), "configs", json_file_name)
|
os.path.dirname(os.path.realpath(__file__)), "configs", json_file_name)
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user