simonduerr commited on
Commit
02a1c15
1 Parent(s): cdd1914

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -81,7 +81,7 @@ def predict_structure(prefix, feature_dict, model_runners, random_seed=0):
81
  f.write(protein.to_pdb(unrelaxed_protein))
82
  return plddts
83
 
84
- @ray.remote(num_gpus=1, max_calls=1)
85
  def run_protgpt2(startsequence, length, repetitionPenalty, top_k_poolsize, max_seqs):
86
  print("running protgpt2")
87
  print(gpu_usage())
@@ -103,7 +103,7 @@ def run_protgpt2(startsequence, length, repetitionPenalty, top_k_poolsize, max_s
103
  #print(gpu_usage())
104
  return sequences
105
 
106
- @ray.remote(num_gpus=1, max_calls=1)
107
  def run_alphafold(startsequence):
108
  print(gpu_usage())
109
  model_runners = {}
 
81
  f.write(protein.to_pdb(unrelaxed_protein))
82
  return plddts
83
 
84
+ @ray.remote(num_gpus=0, max_calls=1)
85
  def run_protgpt2(startsequence, length, repetitionPenalty, top_k_poolsize, max_seqs):
86
  print("running protgpt2")
87
  print(gpu_usage())
 
103
  #print(gpu_usage())
104
  return sequences
105
 
106
+ @ray.remote(num_gpus=0, max_calls=1)
107
  def run_alphafold(startsequence):
108
  print(gpu_usage())
109
  model_runners = {}