Skip to content

Commit fb7f70f

Browse files
authored
Update graph_net.json of HuggingFace transformers models (#247)
* CONTRIBUTE_TUTORIAL_cn.md * Handle big int tensors by converting to sparse COO * Update utils * Update utils * Update utils * Update utils * Update utils * Update paddle test compiler * Add compilation_duration display * resolve conflict * Feat: generate violin figure (benchmark) * revert file * cn->en * Update Analysis * Update * Update * Update * Update * Update * Update * Update * Update * Update * add annotation * Optimized structure, add e2e time * Record 2 types of speedup * update analysis * Update * Update * Update CI * Feat: Record model name & Update config * Update graph_net.json of transformers models * Update * Update * Update
1 parent e6e4b67 commit fb7f70f

File tree

1,487 files changed

+28166
-1484
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

1,487 files changed

+28166
-1484
lines changed

graph_net/torch/extractor.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -70,6 +70,7 @@ def try_rename_placeholder(node):
7070
"num_devices_required": 1,
7171
"num_nodes_required": 1,
7272
"dynamic": bool(dynamic),
73+
"model_name": name,
7374
}
7475
with open(os.path.join(model_path, "graph_net.json"), "w") as f:
7576
json.dump(metadata, f, indent=4)

samples/transformers-auto-model/42dot_LLM-SFT-1.3B/graph_net.json

Lines changed: 19 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,5 +2,23 @@
22
"framework": "torch",
33
"num_devices_required": 1,
44
"num_nodes_required": 1,
5-
"dynamic": false
5+
"dynamic": false,
6+
"model_name": "42dot/42dot_LLM-SFT-1.3B",
7+
"source": "huggingface_hub",
8+
"original_tag": [
9+
"transformers",
10+
"pytorch",
11+
"safetensors",
12+
"llama",
13+
"text-generation",
14+
"causal-lm",
15+
"42dot_llm",
16+
"en",
17+
"ko",
18+
"license:cc-by-nc-4.0",
19+
"autotrain_compatible",
20+
"text-generation-inference",
21+
"endpoints_compatible",
22+
"region:us"
23+
]
624
}

samples/transformers-auto-model/907508196l_wavlm-libri-clean-100h-base-plus-finetuned-ks/graph_net.json

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,5 +2,16 @@
22
"framework": "torch",
33
"num_devices_required": 1,
44
"num_nodes_required": 1,
5-
"dynamic": false
5+
"dynamic": false,
6+
"model_name": "907508196l/wavlm-libri-clean-100h-base-plus-finetuned-ks",
7+
"source": "huggingface_hub",
8+
"original_tag": [
9+
"transformers",
10+
"pytorch",
11+
"tensorboard",
12+
"wavlm",
13+
"audio-classification",
14+
"endpoints_compatible",
15+
"region:us"
16+
]
617
}

samples/transformers-auto-model/AI-Nordics_bert-large-swedish-cased/graph_net.json

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,5 +2,17 @@
22
"framework": "torch",
33
"num_devices_required": 1,
44
"num_nodes_required": 1,
5-
"dynamic": false
5+
"dynamic": false,
6+
"model_name": "AI-Nordics/bert-large-swedish-cased",
7+
"source": "huggingface_hub",
8+
"original_tag": [
9+
"transformers",
10+
"pytorch",
11+
"megatron-bert",
12+
"fill-mask",
13+
"sv",
14+
"autotrain_compatible",
15+
"endpoints_compatible",
16+
"region:us"
17+
]
618
}

samples/transformers-auto-model/AK_ak_nlp/graph_net.json

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,5 +2,17 @@
22
"framework": "torch",
33
"num_devices_required": 1,
44
"num_nodes_required": 1,
5-
"dynamic": false
5+
"dynamic": false,
6+
"model_name": "AK/ak_nlp",
7+
"source": "huggingface_hub",
8+
"original_tag": [
9+
"transformers",
10+
"pytorch",
11+
"jax",
12+
"roberta",
13+
"fill-mask",
14+
"autotrain_compatible",
15+
"endpoints_compatible",
16+
"region:us"
17+
]
618
}

samples/transformers-auto-model/ALINEAR_albert-japanese-v2/graph_net.json

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,5 +2,16 @@
22
"framework": "torch",
33
"num_devices_required": 1,
44
"num_nodes_required": 1,
5-
"dynamic": false
5+
"dynamic": false,
6+
"model_name": "ALINEAR/albert-japanese-v2",
7+
"source": "huggingface_hub",
8+
"original_tag": [
9+
"transformers",
10+
"pytorch",
11+
"albert",
12+
"fill-mask",
13+
"autotrain_compatible",
14+
"endpoints_compatible",
15+
"region:us"
16+
]
617
}

samples/transformers-auto-model/ALINEAR_albert-japanese/graph_net.json

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,5 +2,16 @@
22
"framework": "torch",
33
"num_devices_required": 1,
44
"num_nodes_required": 1,
5-
"dynamic": false
5+
"dynamic": false,
6+
"model_name": "ALINEAR/albert-japanese",
7+
"source": "huggingface_hub",
8+
"original_tag": [
9+
"transformers",
10+
"pytorch",
11+
"albert",
12+
"fill-mask",
13+
"autotrain_compatible",
14+
"endpoints_compatible",
15+
"region:us"
16+
]
617
}

samples/transformers-auto-model/ARTeLab_it5-summarization-fanpage-64/graph_net.json

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,5 +2,19 @@
22
"framework": "torch",
33
"num_devices_required": 1,
44
"num_nodes_required": 1,
5-
"dynamic": false
5+
"dynamic": false,
6+
"model_name": "ARTeLab/it5-summarization-fanpage-64",
7+
"source": "huggingface_hub",
8+
"original_tag": [
9+
"transformers",
10+
"pytorch",
11+
"t5",
12+
"text2text-generation",
13+
"summarization",
14+
"it",
15+
"dataset:ARTeLab/fanpage",
16+
"text-generation-inference",
17+
"endpoints_compatible",
18+
"region:us"
19+
]
620
}

samples/transformers-auto-model/AVSilva_bertimbau-large-fine-tuned-md/graph_net.json

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,5 +2,18 @@
22
"framework": "torch",
33
"num_devices_required": 1,
44
"num_nodes_required": 1,
5-
"dynamic": false
5+
"dynamic": false,
6+
"model_name": "AVSilva/bertimbau-large-fine-tuned-md",
7+
"source": "huggingface_hub",
8+
"original_tag": [
9+
"transformers",
10+
"pytorch",
11+
"bert",
12+
"fill-mask",
13+
"generated_from_trainer",
14+
"license:mit",
15+
"autotrain_compatible",
16+
"endpoints_compatible",
17+
"region:us"
18+
]
619
}

samples/transformers-auto-model/AceInstruct-1.5B/graph_net.json

Lines changed: 20 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,5 +2,24 @@
22
"framework": "torch",
33
"num_devices_required": 1,
44
"num_nodes_required": 1,
5-
"dynamic": false
5+
"dynamic": false,
6+
"model_name": "nvidia/AceInstruct-1.5B",
7+
"source": "huggingface_hub",
8+
"original_tag": [
9+
"safetensors",
10+
"qwen2",
11+
"nvidia",
12+
"AceInstruct",
13+
"code",
14+
"math",
15+
"general_domain",
16+
"instruct_model",
17+
"pytorch",
18+
"text-generation",
19+
"conversational",
20+
"en",
21+
"arxiv:2412.15084",
22+
"license:cc-by-nc-4.0",
23+
"region:us"
24+
]
625
}

0 commit comments

Comments
 (0)