stevengrove commited on
Commit
a722f87
1 Parent(s): e413c25

Update demo with XL model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -21,8 +21,8 @@ from tools.demo import demo
21
  def parse_args():
22
  parser = argparse.ArgumentParser(
23
  description='YOLO-World Demo')
24
- parser.add_argument('--config', default='configs/pretrain/yolo_world_l_t2i_bn_2e-4_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py')
25
- parser.add_argument('--checkpoint', default='yolow-v8_l_clipv2_frozen_t2iv2_bn_o365_goldg_pretrain.pth')
26
  parser.add_argument(
27
  '--work-dir',
28
  help='the directory to save the file containing evaluation metrics')
 
21
  def parse_args():
22
  parser = argparse.ArgumentParser(
23
  description='YOLO-World Demo')
24
+ parser.add_argument('--config', default='configs/pretrain/yolo_world_xl_t2i_bn_2e-4_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py')
25
+ parser.add_argument('--checkpoint', default='yolo_world_v2_xl_obj365v1_goldg_cc3mlite_pretrain.pth')
26
  parser.add_argument(
27
  '--work-dir',
28
  help='the directory to save the file containing evaluation metrics')