Spaces:
Runtime error
Runtime error
File size: 861 Bytes
476ac07 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
# Copyright (c) OpenMMLab. All rights reserved.
import argparse
from xtuner.configs import cfgs_name_path
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument(
'-p', '--pattern', default=None, help='Pattern for fuzzy matching')
args = parser.parse_args()
return args
def main(pattern=None):
args = parse_args()
configs_names = sorted(list(cfgs_name_path.keys()))
print('==========================CONFIGS===========================')
if args.pattern is not None:
print(f'PATTERN: {args.pattern}')
print('-------------------------------')
for name in configs_names:
if args.pattern is None or args.pattern.lower() in name.lower():
print(name)
print('=============================================================')
if __name__ == '__main__':
main()
|