Spaces:
Running
Running
Update inference.py
Browse files- 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 = '
|
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 |
|