Update model/openlamm.py

model/openlamm.py

@@ -160,7 +160,11 @@ class LAMMPEFTModel(nn.Module):
         encoder_pretrain = args['encoder_pretrain'] if 'encoder_pretrain' in args else 'clip'
         self.encoder_pretrain = encoder_pretrain
         assert encoder_pretrain in ['imagebind', 'clip', 'epcl'], f'Encoder_pretrain: {encoder_pretrain} Not Implemented'
-        encoder_ckpt_path = args['encoder_ckpt_path'] if not encoder_pretrain == 'clip' and not os.path.isfile(args['encoder_ckpt_path']) else '~/.cache/clip/ViT-L-14.pt'
+        if not encoder_pretrain == 'clip' or os.path.isfile(args['encoder_ckpt_path']):
+            encoder_ckpt_path = args['encoder_ckpt_path'] 
+        elif not os.path.isfile(args['encoder_ckpt_path']):
+            encoder_ckpt_path = 'ViT-L/14'
+            
         vicuna_ckpt_path = args['vicuna_ckpt_path']
         
         system_header = args['system_header'] if 'system_header' in args else False