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
d0264988
Unverified
Commit
d0264988
authored
Jan 12, 2024
by
Joao Gante
Committed by
GitHub
Jan 12, 2024
Browse files
Generate: deprecate old public functions (#28478)
parent
edb314ae
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
13 additions
and
0 deletions
+13
-0
src/transformers/generation/tf_utils.py
src/transformers/generation/tf_utils.py
+7
-0
src/transformers/generation/utils.py
src/transformers/generation/utils.py
+6
-0
No files found.
src/transformers/generation/tf_utils.py
View file @
d0264988
...
...
@@ -3105,6 +3105,13 @@ def tf_top_k_top_p_filtering(logits, top_k=0, top_p=1.0, filter_value=-float("In
From: https://gist.github.com/thomwolf/1a5a29f6962089e871b94cbd09daf317
"""
warnings
.
warn
(
"`tf_top_k_top_p_filtering` is scheduled for deletion in v4.39. Use `TFTopKLogitsWarper` and "
"`TFTopPLogitsWarper` instead."
,
DeprecationWarning
,
)
logits_shape
=
shape_list
(
logits
)
if
top_k
>
0
:
...
...
src/transformers/generation/utils.py
View file @
d0264988
...
...
@@ -4876,6 +4876,12 @@ def top_k_top_p_filtering(
From: https://gist.github.com/thomwolf/1a5a29f6962089e871b94cbd09daf317
"""
warnings
.
warn
(
"`top_k_top_p_filtering` is scheduled for deletion in v4.39. Use `TopKLogitsWarper` and `TopPLogitsWarper` "
"instead."
,
DeprecationWarning
,
)
if
top_k
>
0
:
logits
=
TopKLogitsWarper
(
top_k
=
top_k
,
filter_value
=
filter_value
,
min_tokens_to_keep
=
min_tokens_to_keep
)(
None
,
logits
...
...
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