diff --git a/src/lambo/examples/run_pretraining.py b/src/lambo/examples/run_pretraining.py index 45531bcc32b48e4b5ce261834522b38271b9d878..a6d0173a33f13e0d3cc7796787ee00e11095e7dd 100644 --- a/src/lambo/examples/run_pretraining.py +++ b/src/lambo/examples/run_pretraining.py @@ -16,8 +16,8 @@ from lambo.learning.train import pretrain from lambo.utils.oscar import read_jsonl_to_documents, download_archive1_from_oscar if __name__=='__main__': - outpath = sys.argv[1] #Path.home() / 'PATH-TO/models/pretrained/' - tmppath = sys.argv[2] #Path.home() / 'PATH-TO/tmp/tmp.jsonl.gz' + outpath = Path(sys.argv[1]) #Path.home() / 'PATH-TO/models/pretrained/' + tmppath = Path(sys.argv[2]) #Path.home() / 'PATH-TO/tmp/tmp.jsonl.gz' # These need to be filled ine before running. OSCAR is avaialable on request. OSCAR_LOGIN = sys.argv[3] OSCAR_PASSWORD = sys.argv[4]