Measure checkpoints above 1B on GPU in bfloat16 only
Browse files- integration.json +9 -1
integration.json
CHANGED
|
@@ -51,7 +51,15 @@
|
|
| 51 |
"download_mb": 20000,
|
| 52 |
"devices": [
|
| 53 |
"cuda"
|
| 54 |
-
]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 55 |
}
|
| 56 |
}
|
| 57 |
}
|
|
|
|
| 51 |
"download_mb": 20000,
|
| 52 |
"devices": [
|
| 53 |
"cuda"
|
| 54 |
+
],
|
| 55 |
+
"variants": {
|
| 56 |
+
"default": {
|
| 57 |
+
"devices": []
|
| 58 |
+
},
|
| 59 |
+
"eager": {
|
| 60 |
+
"devices": []
|
| 61 |
+
}
|
| 62 |
+
}
|
| 63 |
}
|
| 64 |
}
|
| 65 |
}
|