a }Gc@sfddlZddlZddlZddlZddlmZmZddlmZddl m Z ddl m Z m Z d ddZdS) N) defaultdict OrderedDict)DEVICE)ColBERT) print_messageload_checkpointTcCsVtjd|j|j|j|j|jd}|t}t d|dt |j ||d}| ||fS)Nzbert-base-uncased) query_maxlen doc_maxlendimZsimilarity_metricmask_punctuationz#> Loading model checkpoint.) condition)do_print) rfrom_pretrainedrr r similarityr torrr checkpointeval)argsr colbertrr5/home/gupo/~/ColBERT/colbert/evaluation/load_model.py load_model s  r)T)osujsontorchrandom collectionsrrcolbert.parametersrcolbert.modeling.colbertrcolbert.utils.utilsrrrrrrrs