weijiawu commited on
Commit
1899131
1 Parent(s): fc0c525

Update caption_anything.py

Browse files
Files changed (1) hide show
  1. caption_anything.py +1 -1
caption_anything.py CHANGED
@@ -62,7 +62,7 @@ def parse_augment():
62
  parser.add_argument('--clip_filter', action="store_true", help="use clip to filter bad captions")
63
  parser.add_argument('--context_captions', action="store_true", help="use surrounding captions to enhance current caption (TODO)")
64
  parser.add_argument('--disable_regular_box', action="store_true", default = False, help="crop image with a regular box")
65
- parser.add_argument('--device', type=str, default="cuda:0")
66
  parser.add_argument('--port', type=int, default=6086, help="only useful when running gradio applications")
67
  parser.add_argument('--debug', action="store_true")
68
  parser.add_argument('--gradio_share', action="store_true")
 
62
  parser.add_argument('--clip_filter', action="store_true", help="use clip to filter bad captions")
63
  parser.add_argument('--context_captions', action="store_true", help="use surrounding captions to enhance current caption (TODO)")
64
  parser.add_argument('--disable_regular_box', action="store_true", default = False, help="crop image with a regular box")
65
+ parser.add_argument('--device', type=str, default="cpu")
66
  parser.add_argument('--port', type=int, default=6086, help="only useful when running gradio applications")
67
  parser.add_argument('--debug', action="store_true")
68
  parser.add_argument('--gradio_share', action="store_true")