Skip to content

Commit 70489db

Browse files
Merge pull request #45 from RadoslawPlawecki/solve-issue-37
Resolve the issue of padding side set when using Transformers
2 parents 18c643e + f8e8326 commit 70489db

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

llmsql/inference/inference_transformers.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -113,6 +113,7 @@ def inference_transformers(
113113
model_or_model_name_or_path,
114114
token=hf_token,
115115
trust_remote_code=True,
116+
padding_side="left"
116117
)
117118
else:
118119
raise ValueError("Tokenizer must be provided if model is passed directly.")
@@ -121,6 +122,7 @@ def inference_transformers(
121122
tokenizer_or_name,
122123
token=hf_token,
123124
trust_remote_code=True,
125+
padding_side="left"
124126
)
125127
else:
126128
tokenizer = tokenizer_or_name

0 commit comments

Comments
 (0)