File size: 2,280 Bytes
3b4a025
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
105
{
  "model": "meta-encoder-mlx-4bit",
  "load_seconds": 4.304674542014254,
  "embedding_dim": 6656,
  "checks": [
    {
      "name": "text_animal",
      "seconds": 3.765016416000435,
      "scores": [
        0.3861164450645447,
        0.2507832944393158,
        0.19388027489185333
      ],
      "candidates": [
        "cat",
        "dog",
        "horse"
      ],
      "predicted": "cat",
      "expected": "cat",
      "correct": true,
      "norms": [
        1.0,
        1.0,
        1.0,
        0.9999999403953552
      ]
    },
    {
      "name": "text_action",
      "seconds": 1.8200458750070538,
      "scores": [
        0.6188708543777466,
        0.17132438719272614,
        0.19522921741008759
      ],
      "candidates": [
        "Click Cancel order",
        "Open profile settings",
        "Search products"
      ],
      "predicted": "Click Cancel order",
      "expected": "Click Cancel order",
      "correct": true,
      "norms": [
        1.0,
        1.0,
        1.0,
        1.0
      ]
    },
    {
      "name": "image_color",
      "seconds": 1.9545344169891905,
      "scores": [
        0.4928099513053894,
        0.4084464907646179,
        0.40224990248680115
      ],
      "candidates": [
        "red",
        "blue",
        "green"
      ],
      "predicted": "red",
      "expected": "red",
      "correct": true,
      "norms": [
        1.0,
        1.0,
        1.0,
        1.0
      ]
    },
    {
      "name": "video_color",
      "seconds": 1.4857492909941357,
      "scores": [
        0.5262743830680847,
        0.4340401291847229,
        0.418895423412323
      ],
      "candidates": [
        "red",
        "blue",
        "green"
      ],
      "predicted": "red",
      "expected": "red",
      "correct": true,
      "norms": [
        1.0,
        1.0,
        1.0,
        1.0
      ]
    }
  ],
  "total_seconds": 13.330337374994997,
  "peak_mlx_memory_gb": 19.696706988,
  "bf16_mlx_reference_task_cosine": 0.97298264503479,
  "bf16_mlx_reference_video_task_cosine": 0.9559712409973145,
  "reference_scope": "One text task and one two-frame video task compared with streamed BF16 MLX inference from the original checkpoint. Full-model Transformers parity and benchmark accuracy were not measured."
}