Simon Duerr commited on
Commit
907900a
1 Parent(s): 3279dbd
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -50,7 +50,7 @@ sys.path.append("/home/user/app/ProteinMPNN/vanilla_proteinmpnn")
50
  sys.path.append("/home/duerr/phd/08_Code/ProteinMPNN/ProteinMPNN/vanilla_proteinmpnn")
51
 
52
 
53
- tf.config.set_visible_devices([], "GPU")
54
 
55
 
56
  def chain_break(idx_res, Ls, length=200):
@@ -197,7 +197,7 @@ def save_pdb(outs, filename, LEN):
197
  f.write(pdb_lines)
198
 
199
 
200
- @ray.remote(num_gpus=1, max_calls=1)
201
  def run_alphafold(sequences, num_recycles):
202
  recycles = num_recycles
203
  RUNNER, OPT = setup_af(sequences[0])
 
50
  sys.path.append("/home/duerr/phd/08_Code/ProteinMPNN/ProteinMPNN/vanilla_proteinmpnn")
51
 
52
 
53
+ # tf.config.set_visible_devices([], "GPU")
54
 
55
 
56
  def chain_break(idx_res, Ls, length=200):
 
197
  f.write(pdb_lines)
198
 
199
 
200
+ @ray.remote(num_gpus=0, max_calls=1)
201
  def run_alphafold(sequences, num_recycles):
202
  recycles = num_recycles
203
  RUNNER, OPT = setup_af(sequences[0])