lucapinello/chorus-enformer

Chorus mirror: Enformer (DeepMind)

Mirror of the Enformer SavedModel originally hosted at https://tfhub.dev/deepmind/enformer/1 (now redirecting to Kaggle Models). Maintained by chorus for resilience — the canonical source has been migrating across hosts (TFHub → Kaggle).

Source

• Original release: TFHub deepmind/enformer/1 (now redirected to Kaggle).
• Paper: Avsec et al. 2021, Effective gene expression prediction from sequence by integrating long-range interactions, Nature Methods 18, 1196–1203. https://doi.org/10.1038/s41592-021-01252-x
• Reference repo: https://github.com/google-deepmind/deepmind-research/tree/master/enformer

License & attribution

The weights were copied from the SavedModel published by Google DeepMind. Those weights were created by Google DeepMind and are the property of Google LLC.

The model code (the SavedModel graph + the chorus.oracles.enformer loader) is released under the Apache 2.0 license, consistent with the original Enformer release.

These licensing terms are consistent with the terms for the reference code and the model weights.

Files

• saved_model.pb — TensorFlow SavedModel graph
• variables/variables.data-00000-of-00001 — model weights
• variables/variables.index — variables index

Total: ~961 MB.

Usage

In chorus ≥ 0.4 the EnformerOracle fetches from this HF repo by default and falls back to TFHub if HF is unreachable.

from huggingface_hub import snapshot_download
import tensorflow as tf

local = snapshot_download(repo_id="lucapinello/chorus-enformer", repo_type="model")
model = tf.saved_model.load(local)