Skip to content

Commit 04af35c

Browse files
fangfangssjfangfangssj
authored
[New Sample] Add Some NLP Model Computational Graph (#104)
* add nlp model * fix
1 parent e0b8908 commit 04af35c

24 files changed

+27935
-0
lines changed

‎samples/transformers-auto-model/bloom-560m/graph_hash.txt

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

‎samples/transformers-auto-model/bloom-560m/graph_net.json

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/bloom-560m/input_meta.py

Whitespace-only changes.

‎samples/transformers-auto-model/bloom-560m/input_tensor_constraints.py

Whitespace-only changes.

‎samples/transformers-auto-model/bloom-560m/model.py

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

0 commit comments

Comments
 (0)