LTEnjoy commited on
Commit
c622cff
·
verified ·
1 Parent(s): 2f2ecfa

Update demo/modules/init_model.py

Browse files
Files changed (1) hide show
  1. demo/modules/init_model.py +1 -1
demo/modules/init_model.py CHANGED
@@ -10,7 +10,7 @@ from utils.constants import sequence_level
10
  from model.ProTrek.protrek_trimodal_model import ProTrekTrimodalModel
11
  from tqdm import tqdm
12
 
13
- os.system("ls /data/ProTrek_650M_UniRef50")
14
  def load_model():
15
  model_config = {
16
  "protein_config": glob.glob(f"{config.model_dir}/esm2_*")[0],
 
10
  from model.ProTrek.protrek_trimodal_model import ProTrekTrimodalModel
11
  from tqdm import tqdm
12
 
13
+ os.system("ls /data")
14
  def load_model():
15
  model_config = {
16
  "protein_config": glob.glob(f"{config.model_dir}/esm2_*")[0],