From a9dbb28926df5ca750bc0ed8b5722dc7b05336fa Mon Sep 17 00:00:00 2001 From: yiliu30 Date: Wed, 1 Nov 2023 13:13:00 +0800 Subject: [PATCH] remove comments --- src/xturing/engines/causal.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/xturing/engines/causal.py b/src/xturing/engines/causal.py index e8cd001..8f6b7e8 100644 --- a/src/xturing/engines/causal.py +++ b/src/xturing/engines/causal.py @@ -76,7 +76,7 @@ def __init__( model_name, quantization_config=woq_config, trust_remote_code=trust_remote_code, - use_llm_runtime=False, # TODO disable llm runtime for gpt2, removed it later + use_llm_runtime=False, **kwargs) logger.info("Loaded int8 model from Itrex.") else: