Skip to content

Commit a94c43b

Browse files
authored
[New Sample] Add Some NLP Model Computational Graph (#105)
* add nlp model * fix
1 parent 7c02140 commit a94c43b

18 files changed

+12787
-0
lines changed
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
852cea4ff0a26014e69c23284ac770139229c5bc0086b85fbb964909841b1351
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/TinyLlama-1.1B-step-50K-105b/input_meta.py

Whitespace-only changes.

samples/transformers-auto-model/TinyLlama-1.1B-step-50K-105b/input_tensor_constraints.py

Whitespace-only changes.

samples/transformers-auto-model/TinyLlama-1.1B-step-50K-105b/model.py

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

0 commit comments

Comments
 (0)