Skip to content

Commit d05b98c

Browse files
committed
squash! examples : add model conversion tool/example
Update transformer dep to use non-dev version. And also import `AutoModelForCausalLM` instead of `AutoModel` to ensure compatibility with the latest version.
1 parent 695344b commit d05b98c

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
torch~=2.6.0
22
torchvision~=0.21.0
3-
transformers~=4.54.0.dev0
4-
huggingface-hub~=0.33.4
3+
transformers~=4.55.0
4+
huggingface-hub~=0.34.0

examples/model-conversion/scripts/causal/run-casual-gen-embeddings-org.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ import sys
77
import torch
88
import numpy as np
99

10-
from transformers import AutoTokenizer, AutoConfig, AutoModel
10+
from transformers import AutoTokenizer, AutoConfig, AutoModel, AutoModelForCausalLM
1111
from pathlib import Path
1212

1313
unreleased_model_name = os.getenv('UNRELEASED_MODEL_NAME')

0 commit comments

Comments
 (0)