lhallee commited on
Commit
2a6b5ef
·
verified ·
1 Parent(s): 20c9c85

Improve generated model cards

Browse files

Add-only FastPLMs files-only publication. Checkpoint weights and complete-artifact attestations are unchanged.

Files changed (1) hide show
  1. README.md +44 -43
README.md CHANGED
@@ -8,28 +8,21 @@ tags:
8
 
9
  <!-- Generated from src/fastplms/models.toml. Do not edit. -->
10
 
11
- # Synthyra/ESM2-8M
12
 
13
- This checkpoint contains the FastPLMs `ESM2` implementation.
14
 
15
- Accepted inputs are amino-acid sequences tokenized to residue IDs.
16
- Supported Transformers entry points are `AutoConfig`, `AutoModel`,
17
- `AutoModelForMaskedLM`, `AutoModelForSequenceClassification`,
18
- `AutoModelForTokenClassification`.
19
 
20
- ## Capabilities
 
 
21
 
22
- | Feature | Status |
23
- | --- | --- |
24
- | Sequence classification | Supported: base weights with an untrained task head |
25
- | Token classification | Supported: base weights with an untrained task head |
26
- | PEFT fine-tuning | Supported pattern: preserve the separately trained `classifier` |
27
- | Embeddings | Supported: shared ordered embedding API |
28
- | Test-time training | Supported: low-rank masked-residue adaptation |
29
- | Attention variants | Supported: `eager`, `sdpa`, `flex_attention`, `flash_attention_2`, `flash_attention_3` |
30
- | Compliance | Declared: exact release evidence is required |
31
-
32
- A supported interface is not a pretrained downstream predictor. Classification heads start untrained. Compliance metadata does not show that a local build passed its release gate.
33
 
34
  ## Install and platform requirements
35
 
@@ -43,9 +36,14 @@ python -m pip install -r \
43
  The FastPLMs implementation itself is embedded in the model repository.
44
  Transformers loads it through `trust_remote_code=True`.
45
 
46
- This model requires Python 3.11-3.14, PyTorch 2.13, and Transformers 5.13. The artifact requirements include the FlashAttention loader dependency. FlashAttention also requires compatible CUDA hardware and BF16 execution. The Hub quick start needs network access for
47
- the first download. For an air-gapped run, build the manifest-pinned local
48
- artifact first and use the offline example.
 
 
 
 
 
49
 
50
  ## Quick start
51
 
@@ -63,17 +61,17 @@ model = AutoModel.from_pretrained(
63
  For offline validation, replace `model_id` with the manifest-built
64
  `dist/hub/ESM2-8M` path. Pass `local_files_only=True`.
65
 
66
- ## Attention and compliance
 
 
 
 
 
 
67
 
68
- The quick start selects `sdpa` explicitly. Declared variants are `eager`, `sdpa`, `flex_attention`, `flash_attention_2`,
69
- `flash_attention_3`. An unavailable requested backend raises. It does not
70
- silently change implementation.
71
  `output_attentions=True` can use the documented one-call eager fallback to
72
  materialize attention tensors. The configured backend does not change.
73
 
74
- This family declares the `compliance` tier. Release evidence identifies the
75
- checkpoint, backend, dtype, hardware, inputs, and reference revision.
76
-
77
  ## Tokenization and forward inference
78
 
79
  Load the tokenizer from the same artifact as the model. The attention mask
@@ -249,24 +247,28 @@ Plain `AutoModel` omits the optional ESM pooler because this masked-language-
249
  model checkpoint has no trained pooler weights. Pass `add_pooling_layer=True`
250
  only when you intend to initialize and train that head.
251
 
252
- ## Runtime contract
253
 
254
- - Public input: Amino-acid sequences tokenized to residue IDs
255
- - Advertised AutoClasses: `AutoConfig`, `AutoModel`, `AutoModelForMaskedLM`, `AutoModelForSequenceClassification`, `AutoModelForTokenClassification`
256
- - AutoClass weight status: `AutoConfig` = `FastPLMs extension`, `AutoModel` = `pretrained`, `AutoModelForMaskedLM` = `pretrained`, `AutoModelForSequenceClassification` = `base weights + untrained task head`, `AutoModelForTokenClassification` = `base weights + untrained task head`
257
- - Attention implementations: `eager`, `sdpa`, `flex_attention`, `flash_attention_2`, `flash_attention_3`
258
- - Precision policies: `default`
259
  - BF16 execution: `fp32_parameters_autocast`
260
  - Generation contract: `not_applicable`
261
- - Artifact dependency set: `core`
262
  - Weight publication allowed: `true`
263
  - Weight license status: `resolved`
264
  - Redistributable: `true`
265
  - Complete weight publication required: `false`
266
 
267
- ## Release record
 
 
 
 
268
 
269
- - FastPLMs weights: `Synthyra/ESM2-8M`
270
  - Runtime revision: recorded separately in the built artifact and published commit
271
  - Runtime source identities: recorded in `source-record.json`
272
  - Official checkpoint: `facebook/esm2_t6_8M_UR50D`
@@ -276,14 +278,13 @@ only when you intend to initialize and train that head.
276
  - Release tiers: `check`, `compliance`, `feature`, `artifact`, `benchmark`
277
  - Unresolved required file identities: `0`
278
 
279
- The source record records exact file identities, conversion, source revisions,
280
- legal texts, schema, and attestations. A nonzero unresolved count blocks a release.
281
-
282
- ## Validation boundary
283
 
284
  Declared tiers compare configuration, tokenizer behavior, state, and
285
- representative inference with the pinned reference. Metadata does not show that
286
- a build passed, that a backend is faster, or that an output is biologically valid.
 
287
 
288
  ## License
289
 
 
8
 
9
  <!-- Generated from src/fastplms/models.toml. Do not edit. -->
10
 
11
+ # ESM2-8M
12
 
13
+ ## Model overview
14
 
15
+ `Synthyra/ESM2-8M` packages the `facebook/esm2_t6_8M_UR50D` checkpoint with the
16
+ FastPLMs runtime for Hugging Face Transformers. It accepts amino-acid sequences
17
+ tokenized to residue IDs.
 
18
 
19
+ The repository uses the standard Transformers loading interface with
20
+ `trust_remote_code=True`. See Technical details for each registered class and
21
+ whether its weights come from the checkpoint.
22
 
23
+ The sequence- and token-classification classes reuse the pretrained backbone,
24
+ but their task heads are newly initialized. Fine-tune those heads before
25
+ interpreting their logits as predictions.
 
 
 
 
 
 
 
 
26
 
27
  ## Install and platform requirements
28
 
 
36
  The FastPLMs implementation itself is embedded in the model repository.
37
  Transformers loads it through `trust_remote_code=True`.
38
 
39
+ This model requires Python 3.11-3.14, PyTorch 2.13, and Transformers 5.13.
40
+
41
+ The artifact requirements include the FlashAttention loader dependency.
42
+ FlashAttention also requires compatible CUDA hardware and BF16 execution.
43
+
44
+ The Hub quick start needs network access for the first download. For an
45
+ air-gapped run, build the manifest-pinned local artifact first and use the
46
+ offline example.
47
 
48
  ## Quick start
49
 
 
61
  For offline validation, replace `model_id` with the manifest-built
62
  `dist/hub/ESM2-8M` path. Pass `local_files_only=True`.
63
 
64
+ ## Attention backends
65
+
66
+ The quick start uses `sdpa`.
67
+
68
+ Available backends are `eager`, `sdpa`, `flex_attention`, `flash_attention_2`,
69
+ `flash_attention_3`. Requesting an unavailable backend raises instead of
70
+ silently changing implementation.
71
 
 
 
 
72
  `output_attentions=True` can use the documented one-call eager fallback to
73
  materialize attention tensors. The configured backend does not change.
74
 
 
 
 
75
  ## Tokenization and forward inference
76
 
77
  Load the tokenizer from the same artifact as the model. The attention mask
 
247
  model checkpoint has no trained pooler weights. Pass `add_pooling_layer=True`
248
  only when you intend to initialize and train that head.
249
 
250
+ ## Technical details
251
 
252
+ - Inputs: Amino-acid sequences tokenized to residue IDs
253
+ - Transformers classes: `AutoConfig`, `AutoModel`, `AutoModelForMaskedLM`, `AutoModelForSequenceClassification`, `AutoModelForTokenClassification`
254
+ - Checkpoint weights: `AutoConfig` = `FastPLMs extension`, `AutoModel` = `pretrained`, `AutoModelForMaskedLM` = `pretrained`, `AutoModelForSequenceClassification` = `base weights + untrained task head`, `AutoModelForTokenClassification` = `base weights + untrained task head`
255
+ - Attention backends: `eager`, `sdpa`, `flex_attention`, `flash_attention_2`, `flash_attention_3`
256
+ - Precision: `default`
257
  - BF16 execution: `fp32_parameters_autocast`
258
  - Generation contract: `not_applicable`
259
+ - Dependencies: `core`
260
  - Weight publication allowed: `true`
261
  - Weight license status: `resolved`
262
  - Redistributable: `true`
263
  - Complete weight publication required: `false`
264
 
265
+ ## Validation and provenance
266
+
267
+ FastPLMs pins the checkpoint, upstream source revisions, state transformation,
268
+ and required files in `models.toml`. Built artifacts record exact source
269
+ identities and conversion details in `source-record.json`.
270
 
271
+ - FastPLMs checkpoint: `Synthyra/ESM2-8M`
272
  - Runtime revision: recorded separately in the built artifact and published commit
273
  - Runtime source identities: recorded in `source-record.json`
274
  - Official checkpoint: `facebook/esm2_t6_8M_UR50D`
 
278
  - Release tiers: `check`, `compliance`, `feature`, `artifact`, `benchmark`
279
  - Unresolved required file identities: `0`
280
 
281
+ Release validation includes the `compliance` tier. Its evidence identifies the
282
+ checkpoint, backend, dtype, hardware, inputs, and reference revision.
 
 
283
 
284
  Declared tiers compare configuration, tokenizer behavior, state, and
285
+ representative inference with the pinned reference. A nonzero unresolved count
286
+ blocks release. Metadata alone does not show that a build passed, that a backend
287
+ is faster, or that an output is biologically valid.
288
 
289
  ## License
290