From ec19a2a05dfb0f6005a15c7873895d8f153ff5ad Mon Sep 17 00:00:00 2001 From: EricLBuehler Date: Fri, 9 Feb 2024 17:04:41 -0500 Subject: [PATCH] Remove unused --- src/xlora/xlora_utils.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/xlora/xlora_utils.py b/src/xlora/xlora_utils.py index c5b092a..77700bc 100644 --- a/src/xlora/xlora_utils.py +++ b/src/xlora/xlora_utils.py @@ -38,7 +38,6 @@ def load_model( use_flash_attention_2: bool = False, load_xlora: bool = True, verbose: bool = False, - use_cache: bool = False, from_safetensors: bool = True, ) -> Tuple[Union[AutoModelForCausalLM, xLoRAModel], Union[PreTrainedTokenizer, PreTrainedTokenizerFast]]: """