tarekziade HF Staff commited on
Commit
2fe60bc
·
verified ·
1 Parent(s): f3f9d6f

MIC T5-integration (files_sha256 4bacd86fc6c7)

Browse files
Files changed (2) hide show
  1. README.md +2 -2
  2. integration.json +4 -4
README.md CHANGED
@@ -20,8 +20,8 @@ Usage path: AutoTokenizer + AutoModelForSeq2SeqLM.generate, batched.
20
 
21
  | Checkpoint | Slim fixture (every framework PR) |
22
  | --- | --- |
23
- | [`google-t5/t5-small`](https://huggingface.co/google-t5/t5-small) | [`hf-internal-testing/proto-t5-small-slim`](https://huggingface.co/hf-internal-testing/proto-t5-small-slim) |
24
- | [`google/umt5-small`](https://huggingface.co/google/umt5-small) | [`hf-internal-testing/proto-umt5-small-slim`](https://huggingface.co/hf-internal-testing/proto-umt5-small-slim) (real checkpoint unsupported: config.json has no model_type, so AutoModelForSeq2SeqLM cannot load it; the fixture (saved with model_type umt5) covers the UMT5 code path) |
25
 
26
  Checked outputs: `generated_tokens`, `generated_text`, `label_logprobs`, `decoded_roundtrip`, `training.loss`, `training.gradients`, `training.updated`.
27
 
 
20
 
21
  | Checkpoint | Slim fixture (every framework PR) |
22
  | --- | --- |
23
+ | [`google-t5/t5-small`](https://huggingface.co/google-t5/t5-small) | [`hf-internal-testing/mic-t5-small-slim`](https://huggingface.co/hf-internal-testing/mic-t5-small-slim) |
24
+ | [`google/umt5-small`](https://huggingface.co/google/umt5-small) | [`hf-internal-testing/mic-umt5-small-slim`](https://huggingface.co/hf-internal-testing/mic-umt5-small-slim) (real checkpoint unsupported: config.json has no model_type, so AutoModelForSeq2SeqLM cannot load it; the fixture (saved with model_type umt5) covers the UMT5 code path) |
25
 
26
  Checked outputs: `generated_tokens`, `generated_text`, `label_logprobs`, `decoded_roundtrip`, `training.loss`, `training.gradients`, `training.updated`.
27
 
integration.json CHANGED
@@ -5,14 +5,14 @@
5
  "architecture": "T5ForConditionalGeneration, UMT5ForConditionalGeneration",
6
  "checkpoints": {
7
  "google-t5/t5-small": {
8
- "fixture": "hf-internal-testing/proto-t5-small-slim"
9
  },
10
  "google/umt5-small": {
11
- "fixture": "hf-internal-testing/proto-umt5-small-slim"
12
  }
13
  },
14
  "fixtures": {
15
- "hf-internal-testing/proto-t5-small-slim": {
16
  "built_from": "google-t5/t5-small",
17
  "covers": [
18
  "google-t5/t5-small"
@@ -21,7 +21,7 @@
21
  "seed": 17
22
  }
23
  },
24
- "hf-internal-testing/proto-umt5-small-slim": {
25
  "built_from": "google/umt5-small",
26
  "covers": [
27
  "google/umt5-small"
 
5
  "architecture": "T5ForConditionalGeneration, UMT5ForConditionalGeneration",
6
  "checkpoints": {
7
  "google-t5/t5-small": {
8
+ "fixture": "hf-internal-testing/mic-t5-small-slim"
9
  },
10
  "google/umt5-small": {
11
+ "fixture": "hf-internal-testing/mic-umt5-small-slim"
12
  }
13
  },
14
  "fixtures": {
15
+ "hf-internal-testing/mic-t5-small-slim": {
16
  "built_from": "google-t5/t5-small",
17
  "covers": [
18
  "google-t5/t5-small"
 
21
  "seed": 17
22
  }
23
  },
24
+ "hf-internal-testing/mic-umt5-small-slim": {
25
  "built_from": "google/umt5-small",
26
  "covers": [
27
  "google/umt5-small"