Remove the torchao integration (--torchao-config) (#34304)
Co-authored-by: Brayden Zhong <brayden@radixark.ai>
This commit is contained in:
co-authored by
Brayden Zhong
parent
fe0c18effd
commit
5e65dd01a7
@@ -2025,7 +2025,7 @@ class ServerArgs:
|
||||
] = False
|
||||
|
||||
# -------------------------------------------------------------------------
|
||||
# Torch compile and torchao
|
||||
# Torch compile
|
||||
# -------------------------------------------------------------------------
|
||||
enable_torch_compile: A[
|
||||
bool,
|
||||
@@ -2038,12 +2038,6 @@ class ServerArgs:
|
||||
torch_compile_max_bs: A[
|
||||
int, "Set the maximum batch size when using torch compile.", NS("exec.graph")
|
||||
] = 32
|
||||
torchao_config: A[
|
||||
str,
|
||||
"Optimize the model with torchao. Experimental feature. Current choices are: int8dq, int8wo, int4wo-<group_size>, fp8wo, fp8dq-per_tensor, fp8dq-per_row",
|
||||
NS("exec.graph"),
|
||||
] = ""
|
||||
|
||||
# -------------------------------------------------------------------------
|
||||
# Speculative decoding
|
||||
# -------------------------------------------------------------------------
|
||||
|
||||
Reference in New Issue
Block a user