File size: 2,388 Bytes
f3f9d6f
 
 
 
 
 
 
2fe60bc
f3f9d6f
 
2fe60bc
f3f9d6f
 
 
2fe60bc
f3f9d6f
 
 
 
 
 
 
 
2fe60bc
f3f9d6f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
{
  "mic": "0.3",
  "name": "T5 text-to-text generation",
  "usage": "AutoTokenizer + AutoModelForSeq2SeqLM.generate, batched",
  "architecture": "T5ForConditionalGeneration, UMT5ForConditionalGeneration",
  "checkpoints": {
    "google-t5/t5-small": {
      "fixture": "hf-internal-testing/mic-t5-small-slim"
    },
    "google/umt5-small": {
      "fixture": "hf-internal-testing/mic-umt5-small-slim"
    }
  },
  "fixtures": {
    "hf-internal-testing/mic-t5-small-slim": {
      "built_from": "google-t5/t5-small",
      "covers": [
        "google-t5/t5-small"
      ],
      "build": {
        "seed": 17
      }
    },
    "hf-internal-testing/mic-umt5-small-slim": {
      "built_from": "google/umt5-small",
      "covers": [
        "google/umt5-small"
      ],
      "build": {
        "seed": 17,
        "model_class": "UMT5ForConditionalGeneration"
      }
    }
  },
  "integrations": {
    "transformers": {
      "entry_point": "integrations/transformers/usage.md",
      "format": "runnable-markdown",
      "requires": [
        "transformers>=5",
        "torch",
        "sentencepiece"
      ],
      "targets": [
        "fixtures",
        "checkpoints"
      ],
      "unsupported": {
        "google/umt5-small": "config.json has no model_type, so AutoModelForSeq2SeqLM cannot load it; the fixture (saved with model_type umt5) covers the UMT5 code path"
      },
      "variants": {
        "default": {},
        "eager": {
          "attn_implementation": "eager"
        },
        "bfloat16": {
          "dtype": "bfloat16"
        }
      },
      "training": true,
      "resources": {
        "fixture": {
          "memory_gb": 1,
          "download_mb": 150
        },
        "google-t5/t5-small": {
          "memory_gb": 1,
          "download_mb": 242
        }
      }
    }
  },
  "outputs": {
    "runtime": {
      "compare": "exact"
    },
    "generated_tokens": {
      "compare": "exact"
    },
    "generated_text": {
      "compare": "exact"
    },
    "label_logprobs": {
      "compare": "allclose",
      "atol": 0.0001,
      "rtol": 0.0001
    },
    "decoded_roundtrip": {
      "compare": "exact"
    },
    "training.loss": {
      "compare": "allclose",
      "atol": 0.0001,
      "rtol": 0.0001
    },
    "training.gradients": {
      "compare": "all_true"
    },
    "training.updated": {
      "compare": "true"
    }
  }
}