vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

*.py: logging basiconfig refactor to use conditional expression

Browse source
This commit is contained in:
brian khuu 2024-04-18 09:49:20 +10:00
parent 44b058d131
commit 1cc38d81af
7 changed files with 7 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

5
convert-persimmon-to-gguf.py
View file

@ -73,10 +73,7 @@ def main():
parser.add_argument("--adept-inference-dir", type=str, help="path to adept-inference code directory")
parser.add_argument("--verbose", action="store_true", help="increase output verbosity")
args = parser.parse_args()
if args.verbose:
logging.basicConfig(level=logging.DEBUG)
else:
logging.basicConfig(level=logging.INFO)
logging.basicConfig(level=logging.DEBUG if args.verbose else logging.INFO)
sys.path.append(str(args.adept_inference_dir))
persimmon_model = torch.load(args.ckpt_path)
hparams = persimmon_model['args']

5
ggml_vk_generate_shaders.py
View file

@ -2775,10 +2775,7 @@ if __name__ == "__main__":
args = parser.parse_args()
if args.verbose:
logging.basicConfig(level=logging.DEBUG)
else:
logging.basicConfig(level=logging.INFO)
logging.basicConfig(level=logging.DEBUG if args.verbose else logging.INFO)
if args.glslc:
GLSLC = args.glslc

5
gguf-py/scripts/gguf-convert-endian.py
View file

@ -110,10 +110,7 @@ def main() -> None:
args = parser.parse_args(None if len(sys.argv) > 1 else ["--help"])
if args.verbose:
logging.basicConfig(level=logging.DEBUG)
else:
logging.basicConfig(level=logging.INFO)
logging.basicConfig(level=logging.DEBUG if args.verbose else logging.INFO)
logger.info(f'* Loading: {args.model}')
reader = gguf.GGUFReader(args.model, 'r' if args.dry_run else 'r+')

5
gguf-py/scripts/gguf-dump.py
View file

@ -111,10 +111,7 @@ def main() -> None:
args = parser.parse_args(None if len(sys.argv) > 1 else ["--help"])
if args.verbose:
logging.basicConfig(level=logging.DEBUG)
else:
logging.basicConfig(level=logging.INFO)
logging.basicConfig(level=logging.DEBUG if args.verbose else logging.INFO)
if not args.json:
logger.info(f'* Loading: {args.model}')

5
gguf-py/scripts/gguf-set-metadata.py
View file

@ -84,10 +84,7 @@ def main() -> None:
args = parser.parse_args(None if len(sys.argv) > 1 else ["--help"])
if args.verbose:
logging.basicConfig(level=logging.DEBUG)
else:
logging.basicConfig(level=logging.INFO)
logging.basicConfig(level=logging.DEBUG if args.verbose else logging.INFO)
logger.info(f'* Loading: {args.model}')
reader = GGUFReader(args.model, 'r' if args.dry_run else 'r+')

6
tests/test-tokenizer-0-bpe.py
View file

@ -20,11 +20,7 @@ parser.add_argument("--fname-tok", help="path to a text file to tokenize")
parser.add_argument("--verbose", action="store_true", help="increase output verbosity")
args = parser.parse_args()
if args.verbose:
logging.basicConfig(level=logging.DEBUG)
else:
logging.basicConfig(level=logging.INFO)
logging.basicConfig(level=logging.DEBUG if args.verbose else logging.INFO)
dir_tokenizer = args.dir_tokenizer
tokenizer = AutoTokenizer.from_pretrained(dir_tokenizer)

5
tests/test-tokenizer-0-spm.py
View file

@ -21,10 +21,7 @@ parser.add_argument("--verbose", action="store_true", help="increase output verb
args = parser.parse_args()
if args.verbose:
logging.basicConfig(level=logging.DEBUG)
else:
logging.basicConfig(level=logging.INFO)
logging.basicConfig(level=logging.DEBUG if args.verbose else logging.INFO)
dir_tokenizer = args.dir_tokenizer