aksell commited on
Commit
4a42326
1 Parent(s): 2338bb4

Disable ProtGPT2

Browse files

The space crashed.

Files changed (1) hide show
  1. hexviz/app.py +1 -1
hexviz/app.py CHANGED
@@ -15,7 +15,7 @@ Pick a PDB ID, layer and head to visualize attention.
15
 
16
  # Define list of model types
17
  models = [
18
- Model(name=ModelType.ProtGPT2, layers=36, heads=20),
19
  Model(name=ModelType.TAPE_BERT, layers=12, heads=12),
20
  Model(name=ModelType.ZymCTRL, layers=36, heads=16),
21
  # Model(name=ModelType.PROT_T5, layers=24, heads=32),
 
15
 
16
  # Define list of model types
17
  models = [
18
+ # Model(name=ModelType.ProtGPT2, layers=36, heads=20),
19
  Model(name=ModelType.TAPE_BERT, layers=12, heads=12),
20
  Model(name=ModelType.ZymCTRL, layers=36, heads=16),
21
  # Model(name=ModelType.PROT_T5, layers=24, heads=32),