Traceback (most recent call last):
  File "<ipython-input-1-b01b4d15a3e5>", line 651, in <module>
    run_prediction(seq_list, is_prokaryote=is_prokaryote)
  File "<ipython-input-1-b01b4d15a3e5>", line 619, in run_prediction
    predict_structure(feature_dict, multimer, model_name, output_dir)
  File "<ipython-input-1-b01b4d15a3e5>", line 443, in predict_structure
    prediction_result = model_runner.predict(processed_feature_dict, random_seed=0)
  File "/usr/local/lib/python3.7/dist-packages/alphafold/model/model.py", line 167, in predict
    result = self.apply(self.params, jax.random.PRNGKey(random_seed), feat)
  File "/usr/local/lib/python3.7/dist-packages/jax/_src/traceback_util.py", line 162, in reraise_with_filtered_traceback
    return fun(*args, **kwargs)
  File "/usr/local/lib/python3.7/dist-packages/jax/_src/api.py", line 419, in cache_miss
    donated_invars=donated_invars, inline=inline)
  File "/usr/local/lib/python3.7/dist-packages/jax/core.py", line 1632, in bind
    return call_bind(self, fun, *args, **params)
  File "/usr/local/lib/python3.7/dist-packages/jax/core.py", line 1623, in call_bind
    outs = primitive.process(top_trace, fun, tracers, params)
  File "/usr/local/lib/python3.7/dist-packages/jax/core.py", line 1635, in process
    return trace.process_call(self, fun, tracers, params)
  File "/usr/local/lib/python3.7/dist-packages/jax/core.py", line 627, in process_call
    return primitive.impl(f, *tracers, **params)
  File "/usr/local/lib/python3.7/dist-packages/jax/interpreters/xla.py", line 690, in _xla_call_impl
    out = compiled_fun(*args)
  File "/usr/local/lib/python3.7/dist-packages/jax/interpreters/xla.py", line 1100, in _execute_compiled
    out_bufs = compiled.execute(input_bufs)
jax._src.traceback_util.UnfilteredStackTrace: RuntimeError: RESOURCE_EXHAUSTED: Out of memory while trying to allocate 19751361656 bytes.

The stack trace below excludes JAX-internal frames.
The preceding is the original exception that occurred, unmodified.

--------------------

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "<ipython-input-1-b01b4d15a3e5>", line 619, in run_prediction
    predict_structure(feature_dict, multimer, model_name, output_dir)
  File "<ipython-input-1-b01b4d15a3e5>", line 443, in predict_structure
    prediction_result = model_runner.predict(processed_feature_dict, random_seed=0)
  File "/usr/local/lib/python3.7/dist-packages/alphafold/model/model.py", line 167, in predict
    result = self.apply(self.params, jax.random.PRNGKey(random_seed), feat)
  File "/usr/local/lib/python3.7/dist-packages/jax/interpreters/xla.py", line 1100, in _execute_compiled
    out_bufs = compiled.execute(input_bufs)
RuntimeError: RESOURCE_EXHAUSTED: Out of memory while trying to allocate 19751361656 bytes.