Skip to content

Commit 0f454ed

Browse files
committed
fix
1 parent f95efc5 commit 0f454ed

File tree

2 files changed

+29
-5
lines changed

2 files changed

+29
-5
lines changed
Lines changed: 24 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,27 @@
11
samples/ultralytics/yolov3-tinyu
22
samples/torchgeometric/RECT_L
3-
3+
samples/transformers-auto-model/opus-mt-en-gv
4+
samples/timm/convnextv2_base.fcmae_ft_in1k
5+
samples/torchvision/vgg16_bn
6+
samples/timm/regnety_080_tv.tv2_in1k
7+
samples/timm/mobilenetv4_conv_aa_large.e230_r384_in12k
8+
samples/transformers-auto-model/distilbert_distilbert-base-multilingual-cased
9+
samples/transformers-auto-model/OFA-Sys_chinese-clip-vit-large-patch14
10+
samples/transformers-auto-model/opus-mt-tc-bible-big-deu_eng_fra_por_spa-bat
11+
samples/transformers-auto-model/opus-mt-en-tw
12+
samples/timm/hgnet_tiny.paddle_in1k
13+
samples/transformers-auto-model/opus-mt-ase-es
14+
samples/timm/resnetv2_18.ra4e3600r224_in1k
15+
samples/transformers-auto-model/TinyLlama_TinyLlama-1.1B-Chat-v0.4
16+
samples/timm/resnetaa50d.din12k
17+
samples/transformers-auto-model/bge-small-en-v1.5
18+
samples/timm/res2net50_14w_8s.in1k
19+
samples/torchaudio/wavlm_base
20+
samples/transformers-auto-model/opus-mt-en-sal
21+
samples/transformers-auto-model/opus-mt-en-phi
22+
samples/transformers-auto-model/opus-mt-fi-niu
23+
samples/transformers-auto-model/opus-mt-tc-bible-big-gmw-deu_eng_fra_por_spa
24+
samples/transformers-auto-model/opus-mt-NORTHEU-NORTHEU
25+
samples/transformers-auto-model/sentence-transformers_paraphrase-distilroberta-base-v1
26+
samples/transformers-auto-model/TrustSafeAI_RADARVicuna7B
427

graph_net/torch/typical_sequence_split_points.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -156,17 +156,18 @@ def analyze(
156156
input_file = Path(model_paths_file)
157157

158158
with open(input_file, "r") as f:
159-
model_paths = [
160-
Path(model_path_prefix) / line.strip()
159+
rel_model_paths = [
160+
Path(line.strip())
161161
for line in f
162162
if line.strip() and not line.startswith("#")
163163
]
164164

165165
inputs_seqs = []
166166
valid_models = []
167167

168-
for p in model_paths:
169-
seq = self._extract_ops_via_compile(str(p), device)
168+
for p in rel_model_paths:
169+
model_full_path = os.path.join(model_path_prefix, p)
170+
seq = self._extract_ops_via_compile(model_full_path, device)
170171
if seq:
171172
inputs_seqs.append(seq)
172173
valid_models.append((p.name, p))

0 commit comments

Comments
 (0)