Skip to content

Commit a979628

Browse files
committed
Add google/vit-base-patch16-224 computation graph
- Extracted computation graph for google/vit-base-patch16-224 - Validated successfully - Added to transformers-auto-model samples
1 parent f6af6e2 commit a979628

File tree

12 files changed

+5235
-0
lines changed

12 files changed

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

samples/transformers-auto-model/google_vit-base-patch16-224/input_meta.py

Whitespace-only changes.

samples/transformers-auto-model/google_vit-base-patch16-224/input_tensor_constraints.py

Whitespace-only changes.

samples/transformers-auto-model/google_vit-base-patch16-224/model.py

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

samples/transformers-auto-model/google_vit-base-patch16-224/weight_meta.py

Lines changed: 1817 additions & 0 deletions
Large diffs are not rendered by default.
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
c05f6c016eaefabe6f35278d0beaa25f6f3ab32601149736ec2370f368d02566
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
{
2+
"framework": "torch",
3+
"num_devices_required": 1,
4+
"num_nodes_required": 1
5+
}

samples/transformers-auto-model/microsoft_resnet-50/input_meta.py

Whitespace-only changes.

samples/transformers-auto-model/microsoft_resnet-50/input_tensor_constraints.py

Whitespace-only changes.

0 commit comments

Comments
 (0)