wenkai commited on
Commit
285b88f
1 Parent(s): d79b9b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ from esm import pretrained
13
 
14
  model_esm, alphabet = pretrained.load_model_and_alphabet('esm2_t36_3B_UR50D')
15
  model_esm.eval()
16
- model_esm = model.cuda()
17
 
18
  # Load the model
19
  model = Blip2ProteinMistral(config=FAPMConfig(), esm_size='3b')
 
13
 
14
  model_esm, alphabet = pretrained.load_model_and_alphabet('esm2_t36_3B_UR50D')
15
  model_esm.eval()
16
+ model_esm = model_esm.cuda()
17
 
18
  # Load the model
19
  model = Blip2ProteinMistral(config=FAPMConfig(), esm_size='3b')