gabboud commited on
Commit
c767ebc
·
1 Parent(s): f0a8bfb

add bigger ESM models

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,8 +16,8 @@ print("Downloading ESM2 models...")
16
 
17
  MODELS = {
18
  "facebook/esm2_t6_8M_UR50D": "ESM2-8M",
19
- #"facebook/esm2_t12_35M_UR50D": "ESM2-35M",
20
- #"facebook/esm2_t33_650M_UR50D": "ESM2-650M"
21
  }
22
 
23
  cache_dirs = cache_all_models(MODELS)
 
16
 
17
  MODELS = {
18
  "facebook/esm2_t6_8M_UR50D": "ESM2-8M",
19
+ "facebook/esm2_t12_35M_UR50D": "ESM2-35M",
20
+ "facebook/esm2_t33_650M_UR50D": "ESM2-650M"
21
  }
22
 
23
  cache_dirs = cache_all_models(MODELS)