Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
39e14614
"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "4861c25817b5f21da0dc06fd68b7fcb2e7c0ae9a"
Unverified
Commit
39e14614
authored
Jun 10, 2022
by
Domenic Rosati
Committed by
GitHub
Jun 10, 2022
Browse files
fix typo from emtpy to empty (#17643)
parent
13e875cc
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
4 additions
and
4 deletions
+4
-4
src/transformers/generation_beam_constraints.py
src/transformers/generation_beam_constraints.py
+2
-2
src/transformers/generation_logits_process.py
src/transformers/generation_logits_process.py
+1
-1
src/transformers/generation_tf_logits_process.py
src/transformers/generation_tf_logits_process.py
+1
-1
No files found.
src/transformers/generation_beam_constraints.py
View file @
39e14614
...
@@ -139,7 +139,7 @@ class PhrasalConstraint(Constraint):
...
@@ -139,7 +139,7 @@ class PhrasalConstraint(Constraint):
super
(
Constraint
,
self
).
__init__
()
super
(
Constraint
,
self
).
__init__
()
if
not
isinstance
(
token_ids
,
list
)
or
len
(
token_ids
)
==
0
:
if
not
isinstance
(
token_ids
,
list
)
or
len
(
token_ids
)
==
0
:
raise
ValueError
(
f
"`token_ids` has to be a non-em
t
py list, but is
{
token_ids
}
."
)
raise
ValueError
(
f
"`token_ids` has to be a non-emp
t
y list, but is
{
token_ids
}
."
)
if
any
((
not
isinstance
(
token_id
,
int
)
or
token_id
<
0
)
for
token_id
in
token_ids
):
if
any
((
not
isinstance
(
token_id
,
int
)
or
token_id
<
0
)
for
token_id
in
token_ids
):
raise
ValueError
(
f
"Each list in `token_ids` has to be a list of positive integers, but is
{
token_ids
}
."
)
raise
ValueError
(
f
"Each list in `token_ids` has to be a list of positive integers, but is
{
token_ids
}
."
)
...
@@ -271,7 +271,7 @@ class DisjunctiveConstraint(Constraint):
...
@@ -271,7 +271,7 @@ class DisjunctiveConstraint(Constraint):
super
(
Constraint
,
self
).
__init__
()
super
(
Constraint
,
self
).
__init__
()
if
not
isinstance
(
nested_token_ids
,
list
)
or
len
(
nested_token_ids
)
==
0
:
if
not
isinstance
(
nested_token_ids
,
list
)
or
len
(
nested_token_ids
)
==
0
:
raise
ValueError
(
f
"`nested_token_ids` has to be a non-em
t
py list, but is
{
nested_token_ids
}
."
)
raise
ValueError
(
f
"`nested_token_ids` has to be a non-emp
t
y list, but is
{
nested_token_ids
}
."
)
if
any
(
not
isinstance
(
token_ids
,
list
)
for
token_ids
in
nested_token_ids
):
if
any
(
not
isinstance
(
token_ids
,
list
)
for
token_ids
in
nested_token_ids
):
raise
ValueError
(
f
"`nested_token_ids` has to be a list of lists, but is
{
nested_token_ids
}
."
)
raise
ValueError
(
f
"`nested_token_ids` has to be a list of lists, but is
{
nested_token_ids
}
."
)
if
any
(
if
any
(
...
...
src/transformers/generation_logits_process.py
View file @
39e14614
...
@@ -392,7 +392,7 @@ class NoBadWordsLogitsProcessor(LogitsProcessor):
...
@@ -392,7 +392,7 @@ class NoBadWordsLogitsProcessor(LogitsProcessor):
def
__init__
(
self
,
bad_words_ids
:
List
[
List
[
int
]],
eos_token_id
:
int
):
def
__init__
(
self
,
bad_words_ids
:
List
[
List
[
int
]],
eos_token_id
:
int
):
if
not
isinstance
(
bad_words_ids
,
List
)
or
len
(
bad_words_ids
)
==
0
:
if
not
isinstance
(
bad_words_ids
,
List
)
or
len
(
bad_words_ids
)
==
0
:
raise
ValueError
(
f
"`bad_words_ids` has to be a non-em
t
py list, but is
{
bad_words_ids
}
."
)
raise
ValueError
(
f
"`bad_words_ids` has to be a non-emp
t
y list, but is
{
bad_words_ids
}
."
)
if
any
(
not
isinstance
(
bad_word_ids
,
list
)
for
bad_word_ids
in
bad_words_ids
):
if
any
(
not
isinstance
(
bad_word_ids
,
list
)
for
bad_word_ids
in
bad_words_ids
):
raise
ValueError
(
f
"`bad_words_ids` has to be a list of lists, but is
{
bad_words_ids
}
."
)
raise
ValueError
(
f
"`bad_words_ids` has to be a list of lists, but is
{
bad_words_ids
}
."
)
if
any
(
if
any
(
...
...
src/transformers/generation_tf_logits_process.py
View file @
39e14614
...
@@ -297,7 +297,7 @@ class TFNoBadWordsLogitsProcessor(TFLogitsProcessor):
...
@@ -297,7 +297,7 @@ class TFNoBadWordsLogitsProcessor(TFLogitsProcessor):
def
__init__
(
self
,
bad_words_ids
:
List
[
List
[
int
]],
eos_token_id
:
int
):
def
__init__
(
self
,
bad_words_ids
:
List
[
List
[
int
]],
eos_token_id
:
int
):
if
not
isinstance
(
bad_words_ids
,
List
)
or
len
(
bad_words_ids
)
==
0
:
if
not
isinstance
(
bad_words_ids
,
List
)
or
len
(
bad_words_ids
)
==
0
:
raise
ValueError
(
f
"`bad_words_ids` has to be a non-em
t
py list, but is
{
bad_words_ids
}
."
)
raise
ValueError
(
f
"`bad_words_ids` has to be a non-emp
t
y list, but is
{
bad_words_ids
}
."
)
if
any
(
not
isinstance
(
bad_word_ids
,
list
)
for
bad_word_ids
in
bad_words_ids
):
if
any
(
not
isinstance
(
bad_word_ids
,
list
)
for
bad_word_ids
in
bad_words_ids
):
raise
ValueError
(
f
"`bad_words_ids` has to be a list of lists, but is
{
bad_words_ids
}
."
)
raise
ValueError
(
f
"`bad_words_ids` has to be a list of lists, but is
{
bad_words_ids
}
."
)
if
any
(
if
any
(
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment