Measure checkpoints above 1B on GPU in bfloat16 only
Browse files- integration.json +9 -1
integration.json
CHANGED
|
@@ -61,7 +61,15 @@
|
|
| 61 |
"cuda"
|
| 62 |
],
|
| 63 |
"gpu_memory_gb": 8,
|
| 64 |
-
"download_mb": 3000
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 65 |
}
|
| 66 |
}
|
| 67 |
}
|
|
|
|
| 61 |
"cuda"
|
| 62 |
],
|
| 63 |
"gpu_memory_gb": 8,
|
| 64 |
+
"download_mb": 3000,
|
| 65 |
+
"variants": {
|
| 66 |
+
"default": {
|
| 67 |
+
"devices": []
|
| 68 |
+
},
|
| 69 |
+
"eager": {
|
| 70 |
+
"devices": []
|
| 71 |
+
}
|
| 72 |
+
}
|
| 73 |
}
|
| 74 |
}
|
| 75 |
}
|