Skip to content

Commit 4f57655

Browse files
Cutelemon6Cutelemon6
committed
Add google/flan-t5-base computation graph
- Extracted computation graph for google/flan-t5-base - Validated successfully - Added to transformers-auto-model samples
1 parent eb12898 commit 4f57655

File tree

6 files changed

+5102
-0
lines changed

6 files changed

+5102
-0
lines changed

‎samples/transformers-auto-model/google_flan_t5_base/graph_hash.txt‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
9f2c0cadb34e948af7c7f7ac93894654a8c828d0d02048173328e1c332da9a65

‎samples/transformers-auto-model/google_flan_t5_base/graph_net.json‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
{
2+
"framework": "torch",
3+
"num_devices_required": 1,
4+
"num_nodes_required": 1,
5+
"dynamic": false
6+
}

‎samples/transformers-auto-model/google_flan_t5_base/input_meta.py‎

Whitespace-only changes.

‎samples/transformers-auto-model/google_flan_t5_base/input_tensor_constraints.py‎

Whitespace-only changes.

‎samples/transformers-auto-model/google_flan_t5_base/model.py‎

Lines changed: 3367 additions & 0 deletions
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)