Hugging Face's logo

Spaces:
hugging-science
/
ESM2
Running

App Files Community
gabboud commited on
Commit
48ea20b
·
1 Parent(s): 878dd15

fix number of layers in 650M loading

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ print("Downloading ESM2 models...")
17
  MODELS = {
18
  "facebook/esm2_t6_8M_UR50D": "ESM2-8M",
19
  "facebook/esm2_t12_35M_UR50D": "ESM2-35M",
20
- "facebook/esm2_t36_650M_UR50D": "ESM2-650M"
21
  }
22
 
23
  cache_dirs = cache_all_models(MODELS)
 
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)