simonduerr commited on
Commit
489fc19
1 Parent(s): c540203

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -92,9 +92,11 @@ def run_protgpt2(startsequence, length, repetitionPenalty, top_k_poolsize, max_s
92
  eos_token_id=0,
93
  )
94
  print(gpu_usage())
 
 
95
  del protgpt2
96
- torch.cuda.empty_cache()
97
  gc.collect()
 
98
  print(gpu_usage())
99
  return sequences
100
 
 
92
  eos_token_id=0,
93
  )
94
  print(gpu_usage())
95
+ print(sequences)
96
+ print(type(sequences))
97
  del protgpt2
 
98
  gc.collect()
99
+ torch.cuda.empty_cache()
100
  print(gpu_usage())
101
  return sequences
102