Skip to content

Commit cb3e8dd

Browse files
authored
[New Sample] Add google/flan-t5-base, flan-t5-small, vit-large-patch32-224-in21k, vit-large-patch16-224, vit-base-patch32-224-in21k (#110)
* Add google/vit-base-patch32-224-in21k computation graph - Extracted computation graph for google/vit-base-patch32-224-in21k - Validated successfully - Added to transformers-auto-model samples * Add google/flan-t5-small computation graph - Extracted computation graph for google/flan-t5-small - Validated successfully - Added to transformers-auto-model samples * Add google/vit-large-patch16-224 computation graph - Extracted computation graph for google/vit-large-patch16-224 - Validated successfully - Added to transformers-auto-model samples * Add google/vit-large-patch32-224-in21k computation graph * 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 92278f8 commit cb3e8dd

30 files changed

+26690
-0
lines changed
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
9f2c0cadb34e948af7c7f7ac93894654a8c828d0d02048173328e1c332da9a65
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)