chenjoya commited on
Commit
7aff2ad
1 Parent(s): 4cc081a

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -12,7 +12,7 @@ logger = transformers.logging.get_logger('liveinfer')
12
  class LiveInfer:
13
  def __init__(self, ) -> None:
14
  args = parse_args()
15
- args.resume_from_checkpoint = 'checkpoints/live1+_aug_2e'
16
  self.model, self.tokenizer = build_model_and_tokenizer(is_training=False, set_vision_inside=True, **asdict(args))
17
  self.model.to('cuda')
18
 
 
12
  class LiveInfer:
13
  def __init__(self, ) -> None:
14
  args = parse_args()
15
+ args.resume_from_checkpoint = 'live1+_aug_2e/'
16
  self.model, self.tokenizer = build_model_and_tokenizer(is_training=False, set_vision_inside=True, **asdict(args))
17
  self.model.to('cuda')
18