From c74a3cfc982b27edc4ded7a9f66b96d74cd88e7c Mon Sep 17 00:00:00 2001 From: EricLBuehler Date: Tue, 20 Feb 2024 06:05:15 -0500 Subject: [PATCH] Remove unnecessary verbose --- src/xlora/xlora_utils.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/src/xlora/xlora_utils.py b/src/xlora/xlora_utils.py index ae9bbee..bef4f0a 100644 --- a/src/xlora/xlora_utils.py +++ b/src/xlora/xlora_utils.py @@ -88,7 +88,6 @@ def load_model( device_map=device, torch_dtype=dtype, use_flash_attention_2=use_flash_attention_2, - verbose=verbose, ) if hasattr(model.config, "use_cache"): model.config.use_cache = False @@ -96,7 +95,6 @@ def load_model( model_name, trust_remote_code=True, device_map=device, - verbose=verbose, ) tokenizer.pad_token = tokenizer.eos_token tokenizer.padding_side = "right"