Add CONFIG_FORMAT environment variable support
This commit is contained in:
@@ -26,6 +26,7 @@ DEFAULT_ARGS = {
|
|||||||
"trust_remote_code": os.getenv('TRUST_REMOTE_CODE', 'False').lower() == 'true',
|
"trust_remote_code": os.getenv('TRUST_REMOTE_CODE', 'False').lower() == 'true',
|
||||||
"download_dir": os.getenv('DOWNLOAD_DIR', None),
|
"download_dir": os.getenv('DOWNLOAD_DIR', None),
|
||||||
"load_format": os.getenv('LOAD_FORMAT', 'auto'),
|
"load_format": os.getenv('LOAD_FORMAT', 'auto'),
|
||||||
|
"config_format": os.getenv('CONFIG_FORMAT', 'auto'),
|
||||||
"dtype": os.getenv('DTYPE', 'auto'),
|
"dtype": os.getenv('DTYPE', 'auto'),
|
||||||
"kv_cache_dtype": os.getenv('KV_CACHE_DTYPE', 'auto'),
|
"kv_cache_dtype": os.getenv('KV_CACHE_DTYPE', 'auto'),
|
||||||
"quantization_param_path": os.getenv('QUANTIZATION_PARAM_PATH', None),
|
"quantization_param_path": os.getenv('QUANTIZATION_PARAM_PATH', None),
|
||||||
|
|||||||
Reference in New Issue
Block a user